org.nd4j.linalg.convolution

Interface ConvolutionInstance

All Known Implementing Classes:

BaseConvolution, DefaultConvolutionInstance

public interface ConvolutionInstance

Method Summary

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

Method Detail

conv2d

INDArray conv2d(INDArray input,
                INDArray kernel,
                Convolution.Type type)

2d convolution (aka the last 2 dimensions

Parameters:

input - the input to op

kernel - the kernel to convolve with

type -

Returns:

convn

INDArray convn(INDArray input,
               INDArray kernel,
               Convolution.Type type,
               int[] axes)

ND Convolution

Parameters:

input - the input to op

kernel - the kernel to op with

type - the opType of convolution

axes - 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 op

kernel - the kernel to op with

type - the opType of convolution

Returns:

the convolution of the given input and kernel