org.deeplearning4j.nn.layers.convolution

Interface ConvolutionHelper

All Known Implementing Classes:

CudnnConvolutionHelper

public interface ConvolutionHelper

Helper for the convolution layer.

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)
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, ConvolutionMode convolutionMode)
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, ConvolutionMode convolutionMode)

Method Detail

backpropGradient

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,
                                                                     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,
                                               ConvolutionMode convolutionMode)

activate

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