public interface ConvolutionInstance
Modifier and Type | Method and Description |
---|---|
INDArray |
conv2d(INDArray input,
INDArray kernel,
Convolution.Type type)
2d convolution (aka the last 2 dimensions
|
INDArray |
convn(INDArray input,
INDArray kernel,
Convolution.Type type)
ND Convolution
|
INDArray |
convn(INDArray input,
INDArray kernel,
Convolution.Type type,
int[] axes)
ND Convolution
|
INDArray conv2d(INDArray input, INDArray kernel, Convolution.Type type)
input
- the input to opkernel
- the kernel to convolve withtype
- INDArray convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes)
input
- the input to opkernel
- the kernel to op withtype
- the opType of convolutionaxes
- the axes to do the convolution alongINDArray convn(INDArray input, INDArray kernel, Convolution.Type type)
input
- the input to opkernel
- the kernel to op withtype
- the opType of convolutionCopyright © 2019. All rights reserved.