MagmaDNN  1.0
c++NeuralNetworkFramework
flattenlayer.cpp File Reference
Include dependency graph for flattenlayer.cpp:

Functions

template<typename T >
FlattenLayer< T > * magmadnn::layer::flatten (op::Operation< T > *input)
 
template FlattenLayer< int > * magmadnn::layer::flatten (op::Operation< int > *input)
 
template FlattenLayer< float > * magmadnn::layer::flatten (op::Operation< float > *input)
 
template FlattenLayer< double > * magmadnn::layer::flatten (op::Operation< double > *input)
 

Detailed Description

Author
Sedrick Keh
Version
1.0
Date
2019-07-08

Function Documentation

◆ flatten()

template<typename T >
FlattenLayer< T > * magmadnn::layer::flatten ( op::Operation< T > *  input)

A new Flatten layer. Typically used after a convolutional layer. Converts the input tensor into a 2d Tensor by flattening it.

Template Parameters
Tnumeric
Parameters
inputInput to flatten
Returns
FlattenLayer<T>* Flatten layer