pytorch::Branch Class Reference

#include <Branch.hpp>

Inheritance diagram for pytorch::Branch:

Public Member Functions
	Branch (const int &copies)
int	get_copies () const
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	copies

Detailed Description

Todo:

: docs

Constructor & Destructor Documentation

Branch()

pytorch::Branch::Branch ( const int &  copies )

inlineexplicit

Member Function Documentation

forward()

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

get_copies()

int pytorch::Branch::get_copies ( ) const

inline

operator()()

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

copies

int pytorch::Branch::copies

private

---

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

• Branch.hpp