pytorch::Layer Class Reference

#include <Layer.hpp>

Inheritance diagram for pytorch::Layer:

Public Member Functions
virtual std::vector< tensor >	forward (const std::vector< tensor > &input)=0
	Forward function for this layer. More...
virtual std::vector< tensor >	operator() (const std::vector< tensor > &input)=0
	Forward function for this layer. More...

Member Function Documentation

forward()

virtual std::vector<tensor> pytorch::Layer::forward ( const std::vector< tensor > &  input )

inlinepure virtual

Forward function for this layer.

Parameters

input

The input to this layer

Returns

The output of this layer

Implemented in pytorch::BatchNorm2d, pytorch::Conv2d, pytorch::AvgPool2d, pytorch::Linear, pytorch::MaxUnpool2d, pytorch::Softmax, pytorch::ReLU, pytorch::Skip, pytorch::MaxPool2d, pytorch::Hardtanh, pytorch::Tanh, pytorch::Branch, pytorch::Slice, pytorch::Concat, pytorch::Sigmoid, pytorch::Difference, pytorch::Divisor, pytorch::Product, and pytorch::Sum.

operator()()

virtual std::vector<tensor> pytorch::Layer::operator() ( const std::vector< tensor > &  input )

inlinepure virtual

Forward function for this layer.

Parameters

input

The input to this layer

Returns

The output of this layer

Implemented in pytorch::BatchNorm2d, pytorch::Conv2d, pytorch::AvgPool2d, pytorch::Linear, pytorch::MaxUnpool2d, pytorch::Softmax, pytorch::Skip, pytorch::ReLU, pytorch::MaxPool2d, pytorch::Hardtanh, pytorch::Tanh, pytorch::Branch, pytorch::Slice, pytorch::Concat, pytorch::Sigmoid, pytorch::Difference, pytorch::Divisor, pytorch::Product, and pytorch::Sum.

---

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

• Layer.hpp