Class PReLULayer.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.BaseLayer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<PReLULayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.PReLULayer.Builder
-
- Enclosing class:
- PReLULayer
public static class PReLULayer.Builder extends FeedForwardLayer.Builder<PReLULayer.Builder>
-
-
Field Summary
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder
nIn, nOut
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PReLULayerbuild()PReLULayer.BuilderinputShape(long... shape)Explicitly set input shape of incoming activations so that parameters can be initialized properly.PReLULayer.BuildersharedAxes(long... axes)Set the broadcasting axes of PReLU's alpha parameter.-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder
nIn, nIn, nOut, nOut, units
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Method Detail
-
inputShape
public PReLULayer.Builder inputShape(long... shape)
Explicitly set input shape of incoming activations so that parameters can be initialized properly. This explicitly excludes the mini-batch dimension.- Parameters:
shape- shape of input data
-
sharedAxes
public PReLULayer.Builder sharedAxes(long... axes)
Set the broadcasting axes of PReLU's alpha parameter. For instance, given input data of shape [mb, channels, height, width], setting axes to [2,3] will set alpha to shape [channels, 1, 1] and broadcast alpha across height and width dimensions of each channel.- Parameters:
axes- shared/broadcasting axes- Returns:
- Builder
-
build
public PReLULayer build()
- Specified by:
buildin classLayer.Builder<PReLULayer.Builder>
-
-