flattenlayer.cpp File Reference

#include "layer/flatten/flattenlayer.h"

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

Copyright

Copyright (c) 2019

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

T

numeric

Parameters

input

Input to flatten

Returns

FlattenLayer<T>* Flatten layer