Uses of Class
org.deeplearning4j.nn.conf.layers.CapsuleLayer.Builder
-
Packages that use CapsuleLayer.Builder Package Description org.deeplearning4j.nn.conf.layers -
-
Uses of CapsuleLayer.Builder in org.deeplearning4j.nn.conf.layers
Methods in org.deeplearning4j.nn.conf.layers that return CapsuleLayer.Builder Modifier and Type Method Description CapsuleLayer.Builder
CapsuleLayer.Builder. capsuleDimensions(int capsuleDimensions)
Set the number dimensions of each capsuleCapsuleLayer.Builder
CapsuleLayer.Builder. capsules(int capsules)
Set the number of capsules to use.CapsuleLayer.Builder
CapsuleLayer.Builder. hasBias(boolean hasBias)
Sets whether to use bias.CapsuleLayer.Builder
CapsuleLayer.Builder. inputCapsuleDimensions(int inputCapsuleDimensions)
Usually inferred automatically.CapsuleLayer.Builder
CapsuleLayer.Builder. inputCapsules(int inputCapsules)
Usually inferred automatically.CapsuleLayer.Builder
CapsuleLayer.Builder. inputShape(int... inputShape)
Usually inferred automatically.CapsuleLayer.Builder
CapsuleLayer.Builder. routings(int routings)
Set the number of dynamic routing iterations to use.Constructors in org.deeplearning4j.nn.conf.layers with parameters of type CapsuleLayer.Builder Constructor Description CapsuleLayer(CapsuleLayer.Builder builder)
-