pytorch-inference
pytorch::Tanh Class Reference

#include <Activations.hpp>

Inheritance diagram for pytorch::Tanh:
pytorch::Layer

Public Member Functions

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...
 

Detailed Description

Todo:
: docs

Member Function Documentation

◆ forward()

std::vector<tensor> pytorch::Tanh::forward ( const std::vector< tensor > &  input)
inlinevirtual

Forward function for this layer.

Parameters
inputThe 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
inputThe input to this layer
Returns
The output of this layer

Implements pytorch::Layer.


The documentation for this class was generated from the following file: