Package | Description |
---|---|
org.deeplearning4j.nn.conf.inputs | |
org.deeplearning4j.nn.conf.layers | |
org.deeplearning4j.util |
Modifier and Type | Method and Description |
---|---|
static InputType |
InputType.convolutional3D(Convolution3D.DataFormat dataFormat,
long depth,
long height,
long width,
long channels)
Input type for 3D convolutional (CNN3D) 5d data:
If NDHWC format [miniBatchSize, depth, height, width, channels] If NDCWH |
Modifier and Type | Field and Description |
---|---|
protected Convolution3D.DataFormat |
Cnn3DLossLayer.dataFormat |
protected Convolution3D.DataFormat |
Cnn3DLossLayer.Builder.dataFormat |
Modifier and Type | Method and Description |
---|---|
static Convolution3D.DataFormat |
Convolution3D.DataFormat.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static Convolution3D.DataFormat[] |
Convolution3D.DataFormat.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
Convolution3D.Builder |
Convolution3D.Builder.dataFormat(Convolution3D.DataFormat dataFormat) |
Constructor and Description |
---|
Builder(Convolution3D.DataFormat format) |
Modifier and Type | Method and Description |
---|---|
static INDArray |
ConvolutionUtils.reshape2dTo5d(Convolution3D.DataFormat format,
INDArray in2d,
int n,
int d,
int h,
int w,
int ch,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshape5dTo2d(Convolution3D.DataFormat format,
INDArray in,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
static INDArray |
ConvolutionUtils.reshapeCnn3dMask(Convolution3D.DataFormat format,
INDArray mask,
INDArray label,
LayerWorkspaceMgr workspaceMgr,
ArrayType type) |
Copyright © 2018. All rights reserved.