Package 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

Constructor Summary

Constructors

Constructor	Description
BaseConvolution()

Method Summary

Modifier and Type	Method	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

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

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:

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

Returns:

the convolution of the given input and kernel