oa::Flatten

Flatten: Flatten layer with PyTorch-style start_dim/end_dim.

Inheritance

public Module

Public Methods

Matrix oa::Flatten::forward(const Matrix & inInput)

Constructor & Destructor Documentation

oa::Flatten::Flatten( oa::I32 inStartDim = 1, oa::I32 inEndDim = -1 )
No public source comment is attached to this declaration.

Parameters

inStartDim
oa::I32

Default: 1

inEndDim
oa::I32

Default: -1

Public Method Documentation

Matrix oa::Flatten::forward( const Matrix & inInput )
No public source comment is attached to this declaration.

Parameters

inInput
const Matrix &

Returns

Matrix

The declared return value.