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

Functions

template<typename T >
PoolingLayer< T > * magmadnn::layer::pooling (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape, const std::vector< unsigned int > &padding, const std::vector< unsigned int > &strides, pooling_mode mode, bool propagate_nan)
 
template PoolingLayer< int > * magmadnn::layer::pooling (op::Operation< int > *, const std::vector< unsigned int > &, const std::vector< unsigned int > &, const std::vector< unsigned int > &, pooling_mode, bool)
 
template PoolingLayer< float > * magmadnn::layer::pooling (op::Operation< float > *, const std::vector< unsigned int > &, const std::vector< unsigned int > &, const std::vector< unsigned int > &, pooling_mode, bool)
 
template PoolingLayer< double > * magmadnn::layer::pooling (op::Operation< double > *, const std::vector< unsigned int > &, const std::vector< unsigned int > &, const std::vector< unsigned int > &, pooling_mode, bool)
 
template<typename T >
PoolingLayer< T > * magmadnn::layer::pooling (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape, layer::padding_t padding, const std::vector< unsigned int > &strides, pooling_mode mode, bool propagate_nan)
 
template PoolingLayer< int > * magmadnn::layer::pooling (op::Operation< int > *, const std::vector< unsigned int > &, layer::padding_t, const std::vector< unsigned int > &, pooling_mode, bool)
 
template PoolingLayer< float > * magmadnn::layer::pooling (op::Operation< float > *, const std::vector< unsigned int > &, layer::padding_t, const std::vector< unsigned int > &, pooling_mode, bool)
 
template PoolingLayer< double > * magmadnn::layer::pooling (op::Operation< double > *, const std::vector< unsigned int > &, layer::padding_t, const std::vector< unsigned int > &, pooling_mode, bool)
 

Detailed Description

Author
Sedrick Keh
Version
1.0
Date
2019-07-08

Function Documentation

◆ pooling()

template<typename T >
PoolingLayer< T > * magmadnn::layer::pooling ( op::Operation< T > *  input,
const std::vector< unsigned int > &  filter_shape = {2, 2},
const std::vector< unsigned int > &  padding = {0, 0},
const std::vector< unsigned int > &  strides = {1, 1},
pooling_mode  mode = MAX_POOL,
bool  propagate_nan = false 
)

A new Pooling2d layer.

Template Parameters
Tnumeric
Parameters
inputinput, usually output of an activated convolutional layer
filter_shapeshape to filter image
paddinghow to pad pooling
stridesstriding
modeMAX_POOL or AVERAGE_POOL
propagate_nanpropagate nan values
Returns
PoolingLayer<T>* a pooling layer