pytorch::Difference Class Reference

#include <Difference.hpp>

Inheritance diagram for pytorch::Difference:

Public Member Functions
	Difference (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

Difference()

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

inline

Member Function Documentation

forward()

std::vector<tensor> pytorch::Difference::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::Difference::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::Difference::dim

private

n_tensors

int pytorch::Difference::n_tensors

private

---

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

• Difference.hpp