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
|