poolinglayer.h

Go to the documentation of this file.

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

namespace magmadnn {
namespace layer {

template <typename T>
class PoolingLayer : public Layer<T> {
public:
    PoolingLayer(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);

    virtual ~PoolingLayer();

    virtual std::vector<op::Operation<T> *> get_weights();

protected:
    void init();

    pooling_mode mode;
    bool propagate_nan;
    int filter_h, filter_w, pad_h, pad_w, stride_h, stride_w;

};

template <typename T>
PoolingLayer<T>* 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);

/* Indicate pooling type when creating */
template <typename T>
PoolingLayer<T>* pooling(op::Operation<T> *input, const std::vector<unsigned int>& filter_shape={2, 2}, layer::padding_t padding=layer::SAME,
        const std::vector<unsigned int>& strides={1,1}, pooling_mode mode = MAX_POOL, bool propagate_nan=false);

}   // layer
}   // magmadnn