Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers |
Modifier and Type | Method and Description |
---|---|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).
|
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 © 2021. All rights reserved.