poolinglayer.cpp File Reference

#include "layer/pooling/poolinglayer.h"

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

Copyright

Copyright (c) 2019

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

T

numeric

Parameters

input	input, usually output of an activated convolutional layer
filter_shape	shape to filter image
padding	how to pad pooling
strides	striding
mode	MAX_POOL or AVERAGE_POOL
propagate_nan	propagate nan values

Returns

PoolingLayer<T>* a pooling layer