public static class PReLULayer.Builder extends FeedForwardLayer.Builder<PReLULayer.Builder>
nIn, nOut
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
PReLULayer |
build() |
PReLULayer.Builder |
inputShape(long... shape)
Explicitly set input shape of incoming activations so that parameters can be initialized properly.
|
PReLULayer.Builder |
sharedAxes(long... axes)
Set the broadcasting axes of PReLU's alpha parameter.
|
nIn, nIn, nOut, nOut, units
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
public PReLULayer.Builder inputShape(long... shape)
shape
- shape of input datapublic PReLULayer.Builder sharedAxes(long... axes)
axes
- shared/broadcasting axespublic PReLULayer build()
build
in class Layer.Builder<PReLULayer.Builder>
Copyright © 2021. All rights reserved.