poolinglayer.h File Reference

#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"
#include "layer/conv2d/conv2dlayer.h"

Include dependency graph for poolinglayer.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	magmadnn::layer::PoolingLayer< T >

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<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)

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