Class SameDiffLayer.Builder<T extends SameDiffLayer.Builder<T>>
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder<T>
-
- Direct Known Subclasses:
CapsuleLayer.Builder
,CapsuleStrengthLayer.Builder
,LearnedSelfAttentionLayer.Builder
,LocallyConnected1D.Builder
,LocallyConnected2D.Builder
,PrimaryCapsules.Builder
,RecurrentAttentionLayer.Builder
,SelfAttentionLayer.Builder
- Enclosing class:
- SameDiffLayer
public abstract static class SameDiffLayer.Builder<T extends SameDiffLayer.Builder<T>> extends AbstractSameDiffLayer.Builder<T>
-
-
Field Summary
Fields Modifier and Type Field Description protected Map<String,IWeightInit>
paramWeightInit
protected WeightInit
weightInit
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder
biasUpdater, regularization, regularizationBias, updater
-
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 T
weightInit(@NonNull String param, @NonNull IWeightInit weightInit)
T
weightInit(WeightInit weightInit)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder
biasUpdater, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
build, constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Field Detail
-
weightInit
protected WeightInit weightInit
-
paramWeightInit
protected Map<String,IWeightInit> paramWeightInit
-
-
Method Detail
-
weightInit
public T weightInit(WeightInit weightInit)
- Parameters:
weightInit
- Weight initialization to use for the layer
-
weightInit
public T weightInit(@NonNull @NonNull String param, @NonNull @NonNull IWeightInit weightInit)
-
-