org.deeplearning4j.nn.conf.layers

Class GlobalPoolingLayer.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<GlobalPoolingLayer.Builder>

org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer.Builder

Enclosing class:

GlobalPoolingLayer

public static class GlobalPoolingLayer.Builder
extends Layer.Builder<GlobalPoolingLayer.Builder>

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(PoolingType poolingType)

Method Summary

Modifier and Type	Method and Description
GlobalPoolingLayer	build()
GlobalPoolingLayer.Builder	collapseDimensions(boolean collapseDimensions) Whether to collapse dimensions when pooling or not.
GlobalPoolingLayer.Builder	pnorm(int pnorm) P-norm constant.
GlobalPoolingLayer.Builder	poolingDimensions(int... poolingDimensions) Pooling dimensions.
GlobalPoolingLayer.Builder	poolingType(PoolingType poolingType)
void	setPnorm(int pnorm)

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder()

Builder

public Builder(PoolingType poolingType)

Method Detail

poolingDimensions

public GlobalPoolingLayer.Builder poolingDimensions(int... poolingDimensions)

Pooling dimensions. Note: most of the time, this doesn't need to be set, and the defaults can be used. Default for RNN data: pooling dimension 2 (time). Default for CNN data: pooling dimensions 2,3 (height and width) Default for CNN3D data: pooling dimensions 2,3,4 (depth, height and width)

Parameters:

poolingDimensions - Pooling dimensions to use

poolingType

public GlobalPoolingLayer.Builder poolingType(PoolingType poolingType)

Parameters:

poolingType - Pooling type for global pooling

collapseDimensions

public GlobalPoolingLayer.Builder collapseDimensions(boolean collapseDimensions)

Whether to collapse dimensions when pooling or not. Usually you *do* want to do this. Default: true. If true:
- 3d (time series) input with shape [miniBatchSize, vectorSize, timeSeriesLength] -> 2d output [miniBatchSize, vectorSize]
- 4d (CNN) input with shape [miniBatchSize, channels, height, width] -> 2d output [miniBatchSize, channels]
- 5d (CNN3D) input with shape [miniBatchSize, channels, depth, height, width] -> 2d output [miniBatchSize, channels]
If false:
- 3d (time series) input with shape [miniBatchSize, vectorSize, timeSeriesLength] -> 3d output [miniBatchSize, vectorSize, 1]
- 4d (CNN) input with shape [miniBatchSize, channels, height, width] -> 2d output [miniBatchSize, channels, 1, 1]
- 5d (CNN3D) input with shape [miniBatchSize, channels, depth, height, width] -> 2d output [miniBatchSize, channels, 1, 1, 1]

Parameters:

collapseDimensions - Whether to collapse the dimensions or not

pnorm

public GlobalPoolingLayer.Builder pnorm(int pnorm)

P-norm constant. Only used if using PoolingType.PNORM for the pooling type

Parameters:

pnorm - P-norm constant

setPnorm

public void setPnorm(int pnorm)

build

public GlobalPoolingLayer build()

Specified by:

build in class Layer.Builder<GlobalPoolingLayer.Builder>