Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers |
Modifier and Type | Method and Description |
---|---|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.capsuleDimensions(int capsuleDimensions)
Sets the number of dimensions to use in the capsules.
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.capsules(int capsules)
Usually inferred automatically.
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.channels(int channels)
Sets the number of channels to use in the 2d convolution.
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.convolutionMode(ConvolutionMode convolutionMode)
The convolution mode to use in the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.dilation(int... dilation)
Sets the dilation of the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.hasBias(boolean hasBias) |
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.kernelSize(int... kernelSize)
Sets the kernel size of the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.nOut(int nOut)
Sets the number of channels to use in the 2d convolution.
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.padding(int... padding)
Sets the padding of the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.stride(int... stride)
Sets the stride of the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.useLeakyReLU(double leak)
Use a LeakyReLU activation on the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.useReLU()
Use a ReLU activation on the 2d convolution
|
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.useReLU(boolean useRelu)
Whether to use a ReLU activation on the 2d convolution
|
Constructor and Description |
---|
PrimaryCapsules(PrimaryCapsules.Builder builder) |
Copyright © 2022. All rights reserved.