public class DefaultConvolutionInstance extends BaseConvolution
Constructor and Description |
---|
DefaultConvolutionInstance() |
Modifier and Type | Method and Description |
---|---|
INDArray |
convn(INDArray input,
INDArray kernel,
Convolution.Type type,
int[] axes)
ND Convolution
|
conv2d, convn
public 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 alongCopyright © 2020. All rights reserved.