pytorch::Product Class Reference

#include <Product.hpp>

Inheritance diagram for pytorch::Product:

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

Product()

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

inline

Member Function Documentation

forward()

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

private

n_tensors

int pytorch::Product::n_tensors

private

---

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

• Product.hpp