Package org.deeplearning4j.nn.layers.pooling

Class Summary

Class

Description

GlobalPoolingLayer

Global pooling layer - used to do pooling over time for RNNs, and 2d pooling for CNNs. Supports the following PoolingTypes: SUM, AVG, MAX, PNORM Global pooling layer can also handle mask arrays when dealing with variable length inputs. mask arrays are assumed to be 2d, and are fed forward through the network during training or post-training forward pass: - Time series (RNNs, 1d CNNs): mask arrays are shape [miniBatchSize, maxTimeSeriesLength] and contain values 0 or 1 only - CNNs (2d): mask have shape [miniBatchSize, 1, height, 1] or [miniBatchSize, 1, 1, width] or [minibatch, 1, height, width].