| 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.conv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Conv2d operation.
|
SDVariable |
DifferentialFunctionFactory.depthWiseConv2d(SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depthwise Conv2d operation.
|
SDVariable |
DifferentialFunctionFactory.sconv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable Conv2d operation.
|
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
SameDiff.conv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Conv2d operation.
|
SDVariable |
SameDiff.conv2d(String name,
SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Conv2d operation.
|
SDVariable |
SameDiff.depthWiseConv2d(SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depth-wise Conv2d operation.
|
SDVariable |
SameDiff.depthWiseConv2d(String name,
SDVariable[] inputs,
Conv2DConfig depthConv2DConfig)
Depth-wise Conv2d operation.
|
SDVariable |
SameDiff.sconv2d(SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable Conv2d operation.
|
SDVariable |
SameDiff.sconv2d(String name,
SDVariable[] inputs,
Conv2DConfig conv2DConfig)
Separable Conv2d operation.
|
| Modifier and Type | Field and Description |
|---|---|
protected Conv2DConfig |
DepthwiseConv2D.config |
protected Conv2DConfig |
Conv2D.config |
protected Conv2DConfig |
Im2col.conv2DConfig |
protected Conv2DConfig |
Col2Im.conv2DConfig |
| Constructor and Description |
|---|
Col2Im(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputs,
Conv2DConfig conv2DConfig) |
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) |
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.