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 |
Subsampling3DLayer.convolutionMode |
protected ConvolutionMode |
Subsampling3DLayer.BaseSubsamplingBuilder.convolutionMode |
protected ConvolutionMode |
ConvolutionLayer.convolutionMode |
protected ConvolutionMode |
ConvolutionLayer.BaseConvBuilder.convolutionMode |
protected ConvolutionMode |
SubsamplingLayer.convolutionMode |
protected ConvolutionMode |
SubsamplingLayer.BaseSubsamplingBuilder.convolutionMode |
Modifier and Type | Method and Description |
---|---|
T |
Subsampling3DLayer.BaseSubsamplingBuilder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
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.
|
T |
SubsamplingLayer.BaseSubsamplingBuilder.convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
static InputType |
InputTypeUtil.getOutputTypeCnn3DLayers(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
int outputChannels,
int layerIdx,
String layerName,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeCnnLayers(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
int outputDepth,
int layerIdx,
String layerName,
Class<?> layerClass) |
static InputType |
InputTypeUtil.getOutputTypeDeconvLayer(InputType inputType,
int[] kernelSize,
int[] stride,
int[] padding,
int[] dilation,
ConvolutionMode convolutionMode,
int outputDepth,
int layerIdx,
String layerName,
Class<?> layerClass) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
ConvolutionLayer.convolutionMode |
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
ConvolutionHelper.backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
org.nd4j.linalg.api.ndarray.INDArray biasGradView,
org.nd4j.linalg.api.ndarray.INDArray weightGradView,
org.nd4j.linalg.activations.IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
LayerWorkspaceMgr workspaceMgr) |
org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionHelper.preOutput(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.FwdAlgo fwdAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionMode |
Subsampling3DLayer.convolutionMode |
protected ConvolutionMode |
SubsamplingLayer.convolutionMode |
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.api.ndarray.INDArray |
SubsamplingHelper.activate(org.nd4j.linalg.api.ndarray.INDArray input,
boolean training,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
LayerWorkspaceMgr workspaceMgr) |
org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
SubsamplingHelper.backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray epsilon,
int[] kernel,
int[] strides,
int[] pad,
PoolingType poolingType,
ConvolutionMode convolutionMode,
int[] dilation,
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 int[] |
Convolution3DUtils.get3DOutputSize(org.nd4j.linalg.api.ndarray.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 int[] |
ConvolutionUtils.getDeconvolutionOutputSize(org.nd4j.linalg.api.ndarray.INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation)
Get the output size of a deconvolution operation for given input data.
|
static int[] |
ConvolutionUtils.getOutputSize(org.nd4j.linalg.api.ndarray.INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode) |
static int[] |
ConvolutionUtils.getOutputSize(org.nd4j.linalg.api.ndarray.INDArray inputData,
int[] kernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation)
Get the output size (height/width) for the given input data and CNN configuration
|
static void |
ConvolutionUtils.validateConvolutionModePadding(ConvolutionMode mode,
int[] padding)
Check that the convolution mode is consistent with the padding specification
|
static void |
ConvolutionUtils.validateShapes(org.nd4j.linalg.api.ndarray.INDArray inputData,
int[] eKernel,
int[] strides,
int[] padding,
ConvolutionMode convolutionMode,
int[] dilation,
int[] inShape,
boolean atrous) |
Copyright © 2018. All rights reserved.