| Package | Description |
|---|---|
| org.nd4j.autodiff.functions | |
| org.nd4j.autodiff.samediff | |
| org.nd4j.linalg.api.ops.impl.layers.convolution |
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
DifferentialFunctionFactory.col2Im(SDVariable input,
Conv2DConfig config) |
SDVariable |
DifferentialFunctionFactory.conv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Conv2d operation.
|
SDVariable |
DifferentialFunctionFactory.depthWiseConv2d(SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depth-wise Conv2d operation.
|
SDVariable |
DifferentialFunctionFactory.im2Col(SDVariable input,
Conv2DConfig config) |
SDVariable |
DifferentialFunctionFactory.im2ColBp(SDVariable im2colInput,
SDVariable gradientAtOutput,
Conv2DConfig config) |
SDVariable |
DifferentialFunctionFactory.sconv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable Conv2d operation.
|
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
SameDiff.col2Im(SDVariable in,
Conv2DConfig config)
col2im operation for use in 2D convolution operations.
|
SDVariable |
SameDiff.col2Im(String name,
SDVariable in,
Conv2DConfig config)
col2im operation for use in 2D convolution operations.
|
SDVariable |
SameDiff.conv2d(SDVariable[] inputs,
Conv2DConfig config)
2D Convolution operation with optional bias
|
SDVariable |
SameDiff.conv2d(SDVariable layerInput,
SDVariable weights,
Conv2DConfig config)
2D Convolution operation (without bias)
|
SDVariable |
SameDiff.conv2d(SDVariable layerInput,
SDVariable weights,
SDVariable bias,
Conv2DConfig config)
2D Convolution operation with optional bias
|
SDVariable |
SameDiff.conv2d(String name,
SDVariable[] inputs,
Conv2DConfig config)
2D Convolution operation with optional bias
|
SDVariable |
SameDiff.depthWiseConv2d(SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depth-wise convolution 2D operation.
|
SDVariable |
SameDiff.depthWiseConv2d(SDVariable layerInput,
SDVariable depthWeights,
Conv2DConfig config)
Depth-wise 2D convolution operation without bias
|
SDVariable |
SameDiff.depthWiseConv2d(SDVariable layerInput,
SDVariable depthWeights,
SDVariable bias,
Conv2DConfig config)
Depth-wise 2D convolution operation with optional bias
|
SDVariable |
SameDiff.depthWiseConv2d(String name,
SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depth-wise convolution 2D operation.
|
SDVariable |
SameDiff.im2Col(SDVariable in,
Conv2DConfig config)
im2col operation for use in 2D convolution operations.
|
SDVariable |
SameDiff.im2Col(String name,
SDVariable in,
Conv2DConfig config)
im2col operation for use in 2D convolution operations.
|
SDVariable |
SameDiff.sconv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable 2D convolution operation with/without optional bias
|
SDVariable |
SameDiff.sconv2d(String name,
SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable 2D convolution operation with/without optional bias
|
SDVariable |
SameDiff.separableConv2d(SDVariable layerInput,
SDVariable depthWeights,
SDVariable pointWeights,
Conv2DConfig config)
Separable 2D convolution operation without bias
|
SDVariable |
SameDiff.separableConv2d(SDVariable layerInput,
SDVariable depthWeights,
SDVariable pointWeights,
SDVariable bias,
Conv2DConfig config)
Separable 2D convolution operation with optional bias
|
| Modifier and Type | Field and Description |
|---|---|
protected Conv2DConfig |
Conv2D.config |
protected Conv2DConfig |
DepthwiseConv2D.config |
protected Conv2DConfig |
Im2colBp.conv2DConfig |
protected Conv2DConfig |
Im2col.conv2DConfig |
protected Conv2DConfig |
Col2Im.conv2DConfig |
| Constructor and Description |
|---|
Col2Im(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig conv2DConfig) |
Col2Im(SameDiff sd,
SDVariable input,
Conv2DConfig config) |
Conv2D(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig config) |
Conv2DDerivative(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig config) |
DepthwiseConv2D(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig config) |
Im2col(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig conv2DConfig) |
Im2col(SameDiff sd,
SDVariable input,
Conv2DConfig config) |
Im2colBp(SameDiff sd,
SDVariable input,
Conv2DConfig config) |
Im2colBp(SameDiff sameDiff,
SDVariable i2cInput,
SDVariable gradAtOutput,
Conv2DConfig conv2DConfig) |
SConv2D(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig conv2DConfig) |
SConv2DDerivative(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig conv2DConfig) |
Copyright © 2018. All rights reserved.