org.deeplearning4j.nn.conf.layers.samediff

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 and 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 and Description
Builder()

Method Summary

Modifier and Type	Method and 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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

weightInit

protected WeightInit weightInit

paramWeightInit

protected Map<String,IWeightInit> paramWeightInit

Constructor Detail

Builder

public Builder()

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)