org.nd4j.linalg.convolution

Interface ConvolutionInstance

All Known Implementing Classes:

BaseConvolution, DefaultConvolutionInstance

public interface ConvolutionInstance

Convolution instance. Implementations of convolution algorithms

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
INDArray	conv2d(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type)
INDArray	conv2d(INDArray input, INDArray kernel, Convolution.Type type) 2d convolution (aka the last 2 dimensions
IComplexNDArray	convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type) ND Convolution
IComplexNDArray	convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type, int[] axes) ND Convolution
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:

conv2d

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

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 type of convolution

axes - the axes to do the convolution along

Returns:

the convolution of the given input and kernel

convn

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

ND Convolution

Parameters:

input - the input to op

kernel - the kernel to op with

type - the type 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 type of convolution

Returns:

the convolution of the given input and kernel

convn

IComplexNDArray convn(IComplexNDArray input,
                      IComplexNDArray kernel,
                      Convolution.Type type)

ND Convolution

Parameters:

input - the input to op

kernel - the kernel to op with

type - the type of convolution

Returns:

the convolution of the given input and kernel