Package org.nd4j.linalg.convolution
Interface ConvolutionInstance
-
- All Known Implementing Classes:
BaseConvolution
,DefaultConvolutionInstance
public interface ConvolutionInstance
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description INDArray
conv2d(INDArray input, INDArray kernel, Convolution.Type type)
2d convolution (aka the last 2 dimensionsINDArray
convn(INDArray input, INDArray kernel, Convolution.Type type)
ND ConvolutionINDArray
convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes)
ND Convolution
-
-
-
Method Detail
-
conv2d
INDArray conv2d(INDArray input, INDArray kernel, Convolution.Type type)
2d convolution (aka the last 2 dimensions- Parameters:
input
- the input to opkernel
- the kernel to convolve withtype
-- Returns:
-
convn
INDArray convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes)
ND Convolution- Parameters:
input
- the input to opkernel
- the kernel to op withtype
- the opType of convolutionaxes
- the axes to do the convolution along- Returns:
- the convolution of the given input and kernel
-
convn
INDArray convn(INDArray input, INDArray kernel, Convolution.Type type)
ND Convolution- Parameters:
input
- the input to opkernel
- the kernel to op withtype
- the opType of convolution- Returns:
- the convolution of the given input and kernel
-
-