Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
NeuralNetConfiguration.Builder.convolutionMode |
Modifier and Type | Method and Description |
---|---|
static ConvolutionMode |
ConvolutionMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static ConvolutionMode[] |
ConvolutionMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
NeuralNetConfiguration.Builder |
NeuralNetConfiguration.Builder.convolutionMode(ConvolutionMode convolutionMode)
Sets the convolution mode for convolutional layers, which impacts padding and output sizes.
|
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
ConvolutionLayer.convolutionMode |
protected ConvolutionMode |
ConvolutionLayer.BaseConvBuilder.convolutionMode
Set the convolution mode for the Convolution layer.
|
protected ConvolutionMode |
Subsampling3DLayer.convolutionMode |
protected ConvolutionMode |
Subsampling3DLayer.BaseSubsamplingBuilder.convolutionMode
Set the convolution mode for the Convolution layer.
|
protected ConvolutionMode |
SubsamplingLayer.convolutionMode |
protected ConvolutionMode |
SubsamplingLayer.BaseSubsamplingBuilder.convolutionMode
Set the convolution mode for the Convolution layer.
|
Modifier and Type | Method and Description |
---|---|
Convolution3D.Builder |
Convolution3D.Builder.convolutionMode(ConvolutionMode mode) |
T |
ConvolutionLayer.BaseConvBuilder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
Deconvolution2D.Builder |
Deconvolution2D.Builder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
Deconvolution3D.Builder |
Deconvolution3D.Builder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
LocallyConnected1D.Builder |
LocallyConnected1D.Builder.convolutionMode(ConvolutionMode cm) |
LocallyConnected2D.Builder |
LocallyConnected2D.Builder.convolutionMode(ConvolutionMode cm) |
PrimaryCapsules.Builder |
PrimaryCapsules.Builder.convolutionMode(ConvolutionMode convolutionMode)
The convolution mode to use in the 2d convolution
|
T |
Subsampling3DLayer.BaseSubsamplingBuilder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
T |
SubsamplingLayer.BaseSubsamplingBuilder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
static InputType |
InputTypeUtil.getOutputTypeCnn1DLayers(InputType inputType,
int kH,
int sH,
int padH,
int dilation,
ConvolutionMode convolutionMode,
long outputDepth,
long layerIdx,
String layerName,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeCnn3DLayers(InputType inputType,
Convolution3D.DataFormat dataFormat,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
long outputChannels,
long layerIdx,
String layerName,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeCnnLayers(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
long outputDepth,
long layerIdx,
String layerName,
Class<?> layerClass)
|
static InputType |
InputTypeUtil.getOutputTypeCnnLayers(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
long outputDepth,
long layerIdx,
String layerName,
CNN2DFormat format,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeDeconv3dLayer(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
Convolution3D.DataFormat dataFormat,
long outputDepth,
long layerIdx,
String layerName,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeDeconvLayer(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
long outputDepth,
long layerIdx,
String layerName,
Class<?> layerClass) |
protected void |
ConvolutionLayer.BaseConvBuilder.setConvolutionMode(ConvolutionMode convolutionMode) |
protected void |
Subsampling3DLayer.BaseSubsamplingBuilder.setConvolutionMode(ConvolutionMode convolutionMode) |
void |
SubsamplingLayer.BaseSubsamplingBuilder.setConvolutionMode(ConvolutionMode convolutionMode) |
Constructor and Description |
---|
Builder(int capsuleDimensions,
int channels,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
ConvolutionLayer.convolutionMode |
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,INDArray> |
ConvolutionHelper.backpropGradient(INDArray input,
INDArray weights,
INDArray bias,
INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
INDArray biasGradView,
INDArray weightGradView,
IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
INDArray |
ConvolutionHelper.preOutput(INDArray input,
INDArray weights,
INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.FwdAlgo fwdAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
Subsampling3DLayer.convolutionMode |
protected ConvolutionMode |
SubsamplingLayer.convolutionMode |
Modifier and Type | Method and Description |
---|---|
INDArray |
SubsamplingHelper.activate(INDArray input,
boolean training,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Pair<Gradient,INDArray> |
SubsamplingHelper.backpropGradient(INDArray input,
INDArray epsilon,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Method and Description |
---|---|
INDArray |
MKLDNNSubsamplingHelper.activate(INDArray input,
boolean training,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Pair<Gradient,INDArray> |
MKLDNNConvHelper.backpropGradient(INDArray input,
INDArray weights,
INDArray bias,
INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
INDArray biasGradView,
INDArray weightGradView,
IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Pair<Gradient,INDArray> |
MKLDNNSubsamplingHelper.backpropGradient(INDArray input,
INDArray epsilon,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
INDArray |
MKLDNNConvHelper.preOutput(INDArray input,
INDArray weights,
INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.FwdAlgo fwdAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
FineTuneConfiguration.convolutionMode |
Modifier and Type | Method and Description |
---|---|
FineTuneConfiguration.Builder |
FineTuneConfiguration.Builder.convolutionMode(ConvolutionMode convolutionMode)
Sets the convolution mode for convolutional layers, which impacts padding and output sizes.
|
Modifier and Type | Method and Description |
---|---|
static INDArray |
ConvolutionUtils.cnn1dMaskReduction(INDArray in,
int kernel,
int stride,
int padding,
int dilation,
ConvolutionMode cm)
Given a mask array for a 1D CNN layer of shape [minibatch, sequenceLength], reduce the mask according to the 1D CNN layer configuration.
|
static INDArray |
ConvolutionUtils.cnn2dMaskReduction(INDArray inMask,
int[] kernel,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode)
Reduce a 2d CNN layer mask array (of 0s and 1s) according to the layer configuration.
|
static int[] |
Convolution3DUtils.get3DOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation,
boolean isNCDHW)
Get the output size (depth/height/width) for the given input data and CNN3D configuration
|
static long[] |
ConvolutionUtils.getDeconvolution3DOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
Convolution3D.DataFormat dataFormat)
Get the output size of a deconvolution operation for given input data.
|
static int[] |
ConvolutionUtils.getDeconvolutionOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format)
Get the output size of a deconvolution operation for given input data.
|
static int[] |
ConvolutionUtils.getOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode)
Deprecated.
|
static int[] |
ConvolutionUtils.getOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation)
|
static int[] |
ConvolutionUtils.getOutputSize(INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format)
Get the output size (height/width) for the given input data and CNN configuration
|
static int |
Convolution1DUtils.getOutputSize(INDArray inputData,
int kernel,
int strides,
int padding,
ConvolutionMode convolutionMode) |
static int |
Convolution1DUtils.getOutputSize(INDArray inputData,
int kernel,
int strides,
int padding,
ConvolutionMode convolutionMode,
int dilation)
Get the output size (height) for the given input data and CNN1D configuration
|
static long |
Convolution1DUtils.getOutputSize(long inH,
int kernel,
int strides,
int padding,
ConvolutionMode convolutionMode,
int dilation)
Get the output size (height) for the given input data and CNN1D configuration
|
static PaddingMode |
ConvolutionUtils.paddingModeForConvolutionMode(ConvolutionMode convolutionMode)
Convert
ConvolutionMode
to PaddingMode
Same : PaddingMode.SAME
Strict , Truncate : PaddingMode.VALID
Causal : PaddingMode.VALID |
static void |
Convolution1DUtils.validateConvolutionModePadding(ConvolutionMode mode,
int padding)
Check that the convolution mode is consistent with the padding specification
|
static void |
ConvolutionUtils.validateConvolutionModePadding(ConvolutionMode mode,
int[] padding)
Check that the convolution mode is consistent with the padding specification
|
static void |
ConvolutionUtils.validateShapes(INDArray inputData,
int[] eKernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation,
int[] inShape,
boolean atrous) |
static void |
Convolution1DUtils.validateShapes(INDArray inputData,
int eKernel,
int strides,
int padding,
ConvolutionMode convolutionMode,
int dilation,
int inShape,
boolean atrous) |
Copyright © 2021. All rights reserved.