#include "layer/flatten/flattenlayer.h"
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) |
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.
T | numeric |
input | Input to flatten |