org.nd4j.linalg.convolution

Class BaseConvolution

java.lang.Object

org.nd4j.linalg.convolution.BaseConvolution

All Implemented Interfaces:

ConvolutionInstance

Direct Known Subclasses:

DefaultConvolutionInstance

public abstract class BaseConvolution
extends Object
implements ConvolutionInstance

Base convolution implementation

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
BaseConvolution()

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
INDArray	convn(INDArray input, INDArray kernel, Convolution.Type type) ND Convolution

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.linalg.convolution.ConvolutionInstance

convn, convn

Constructor Detail

BaseConvolution

public BaseConvolution()

Method Detail

conv2d

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

2d convolution (aka the last 2 dimensions

Specified by:

conv2d in interface ConvolutionInstance

Parameters:

input - the input to op

kernel - the kernel to convolve with

type -

Returns:

conv2d

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

Specified by:

conv2d in interface ConvolutionInstance

convn

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

ND Convolution

Specified by:

convn in interface ConvolutionInstance

Parameters:

input - the input to transform

kernel - the kernel to transform with

type - the type of convolution

Returns:

the convolution of the given input and kernel

convn

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

ND Convolution

Specified by:

convn in interface ConvolutionInstance

Parameters:

input - the input to transform

kernel - the kernel to transform with

type - the type of convolution

Returns:

the convolution of the given input and kernel