pytorch::Slice Class Reference

#include <Slice.hpp>

Inheritance diagram for pytorch::Slice:

Public Member Functions
	Slice (const int &dim, const int &slices)
int	get_dim () 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	dim
int	slices

Detailed Description

Todo:

: docs

Constructor & Destructor Documentation

Slice()

pytorch::Slice::Slice ( const int &  dim, const int &  slices  )

inline

Member Function Documentation

forward()

std::vector<tensor> pytorch::Slice::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_dim()

int pytorch::Slice::get_dim ( ) const

inline

operator()()

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

private

slices

int pytorch::Slice::slices

private

---

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

• Slice.hpp