pytorch::Concat Class Reference

#include <Concatenate.hpp>

Inheritance diagram for pytorch::Concat:

Public Member Functions
	Concat (const int &dim)
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

Constructor & Destructor Documentation

Concat()

pytorch::Concat::Concat ( const int &  dim )

inlineexplicit

Member Function Documentation

forward()

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

private

---

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

• Concatenate.hpp