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, String 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, String afn, ConvolutionLayer.AlgoMode mode)
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)

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,
                                                                     String afn,
                                                                     ConvolutionLayer.AlgoMode mode)

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)

activate

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