pytorch::Sum Class Reference

#include <Sum.hpp>

Inheritance diagram for pytorch::Sum:

Public Member Functions
	Sum (const int &dim, const int &n_tensors)
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...

Private Attributes
int	dim
int	n_tensors

Constructor & Destructor Documentation

Sum()

pytorch::Sum::Sum ( const int &  dim, const int &  n_tensors  )

inline

Member Function Documentation

forward()

std::vector<tensor> pytorch::Sum::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::Sum::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.

Member Data Documentation

dim

int pytorch::Sum::dim

private

n_tensors

int pytorch::Sum::n_tensors

private

---

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

• Sum.hpp