org.nd4j.linalg.convolution

Class Convolution

java.lang.Object

org.nd4j.linalg.convolution.Convolution

public class Convolution
extends Object

Convolution is the code for applying the convolution operator. http://www.inf.ufpr.br/danielw/pos/ci724/20102/HIPR2/flatjavasrc/Convolution.java

Author:

Adam Gibson

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Convolution.Type

Method Summary

Methods

Modifier and Type	Method and Description
static INDArray	conv2d(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type)
static INDArray	conv2d(INDArray input, INDArray kernel, Convolution.Type type) 2d convolution (aka the last 2 dimensions
static IComplexNDArray	convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type) ND Convolution
static IComplexNDArray	convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type, int[] axes) ND Convolution
static INDArray	convn(INDArray input, INDArray kernel, Convolution.Type type) ND Convolution
static INDArray	convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes) ND Convolution

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

conv2d

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

2d convolution (aka the last 2 dimensions

Parameters:

input - the input to transform

kernel - the kernel to convolve with

type -

Returns:

conv2d

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

convn

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

ND Convolution

Parameters:

input - the input to transform

kernel - the kernel to transform with

type - the type of convolution

axes - the axes to do the convolution along

Returns:

the convolution of the given input and kernel

convn

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

ND Convolution

Parameters:

input - the input to transform

kernel - the kernel to transform with

type - the type of convolution

axes - the axes to do the convolution along

Returns:

the convolution of the given input and kernel

convn

public static INDArray convn(INDArray input,
             INDArray kernel,
             Convolution.Type type)

ND Convolution

Parameters:

input - the input to applyTransformToOrigin

kernel - the kernel to applyTransformToOrigin with

type - the type of convolution

Returns:

the convolution of the given input and kernel

convn

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

ND Convolution

Parameters:

input - the input to applyTransformToOrigin

kernel - the kernel to applyTransformToOrigin with

type - the type of convolution

Returns:

the convolution of the given input and kernel