org.deeplearning4j.nn.layers.convolution

Interface ConvolutionHelper

public interface ConvolutionHelper

Helper for the convolution layer.

Author:

saudet

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	activate(org.nd4j.linalg.api.ndarray.INDArray z, org.nd4j.linalg.activations.IActivation afn)
org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray>	backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray weights, org.nd4j.linalg.api.ndarray.INDArray delta, int[] kernel, int[] strides, int[] pad, org.nd4j.linalg.api.ndarray.INDArray biasGradView, org.nd4j.linalg.api.ndarray.INDArray weightGradView, org.nd4j.linalg.activations.IActivation afn, ConvolutionLayer.AlgoMode mode, ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo, ConvolutionLayer.BwdDataAlgo bwdDataAlgo, ConvolutionMode convolutionMode)
boolean	checkSupported()
org.nd4j.linalg.api.ndarray.INDArray	preOutput(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray weights, org.nd4j.linalg.api.ndarray.INDArray bias, int[] kernel, int[] strides, int[] pad, ConvolutionLayer.AlgoMode mode, ConvolutionLayer.FwdAlgo fwdAlgo, ConvolutionMode convolutionMode)

Method Detail

checkSupported

boolean checkSupported()

backpropGradient

org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray weights,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray delta,
                                                                                                int[] kernel,
                                                                                                int[] strides,
                                                                                                int[] pad,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray biasGradView,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray weightGradView,
                                                                                                org.nd4j.linalg.activations.IActivation afn,
                                                                                                ConvolutionLayer.AlgoMode mode,
                                                                                                ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
                                                                                                ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
                                                                                                ConvolutionMode convolutionMode)

preOutput

org.nd4j.linalg.api.ndarray.INDArray preOutput(org.nd4j.linalg.api.ndarray.INDArray input,
                                               org.nd4j.linalg.api.ndarray.INDArray weights,
                                               org.nd4j.linalg.api.ndarray.INDArray bias,
                                               int[] kernel,
                                               int[] strides,
                                               int[] pad,
                                               ConvolutionLayer.AlgoMode mode,
                                               ConvolutionLayer.FwdAlgo fwdAlgo,
                                               ConvolutionMode convolutionMode)

activate

org.nd4j.linalg.api.ndarray.INDArray activate(org.nd4j.linalg.api.ndarray.INDArray z,
                                              org.nd4j.linalg.activations.IActivation afn)