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 Description Builder()
Builder(PoolingType poolingType)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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
-
-
-
-
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 usingPoolingType.PNORM
for the pooling type- Parameters:
pnorm
- P-norm constant
-
setPnorm
public void setPnorm(int pnorm)
-
build
public GlobalPoolingLayer build()
- Specified by:
build
in classLayer.Builder<GlobalPoolingLayer.Builder>
-
-