#include <Activations.hpp>
|
std::vector< tensor > | forward (const std::vector< tensor > &input) |
| Forward function for this layer. More...
|
|
std::vector< tensor > | operator() (const std::vector< tensor > &input) |
| Forward function for this layer. More...
|
|
◆ forward()
std::vector<tensor> pytorch::Tanh::forward |
( |
const std::vector< tensor > & |
input | ) |
|
|
inlinevirtual |
Forward function for this layer.
- Parameters
-
input | The input to this layer |
- Returns
- The output of this layer
Implements pytorch::Layer.
◆ operator()()
std::vector<tensor> pytorch::Tanh::operator() |
( |
const std::vector< tensor > & |
input | ) |
|
|
inlinevirtual |
Forward function for this layer.
- Parameters
-
input | The input to this layer |
- Returns
- The output of this layer
Implements pytorch::Layer.
The documentation for this class was generated from the following file: