public static class GlobalPoolingLayer.Builder extends Layer.Builder<GlobalPoolingLayer.Builder>
activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit
Constructor and Description |
---|
Builder() |
Builder(PoolingType poolingType) |
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) |
activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit
public Builder()
public Builder(PoolingType poolingType)
public GlobalPoolingLayer.Builder poolingDimensions(int... poolingDimensions)
poolingDimensions
- Pooling dimensions to usepublic GlobalPoolingLayer.Builder poolingType(PoolingType poolingType)
poolingType
- Pooling type for global poolingpublic GlobalPoolingLayer.Builder collapseDimensions(boolean collapseDimensions)
collapseDimensions
- Whether to collapse the dimensions or notpublic GlobalPoolingLayer.Builder pnorm(int pnorm)
PoolingType.PNORM
for the pooling typepnorm
- P-norm constantpublic GlobalPoolingLayer build()
build
in class Layer.Builder<GlobalPoolingLayer.Builder>