Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers |
Modifier and Type | Method and Description |
---|---|
CapsuleLayer.Builder |
CapsuleLayer.Builder.capsuleDimensions(int capsuleDimensions)
Set the number dimensions of each capsule
|
CapsuleLayer.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.
|
Constructor and Description |
---|
CapsuleLayer(CapsuleLayer.Builder builder) |
Copyright © 2022. All rights reserved.