Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers |
Modifier and Type | Method and Description |
---|---|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.convolutionType(org.nd4j.linalg.convolution.Convolution.Type convolutionType)
Deprecated.
Use
convolutionMode |
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.kernelSize(int... kernelSize)
Size of the convolution
rows/columns
|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.padding(int... padding) |
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.stride(int... stride) |
Copyright © 2016. All Rights Reserved.