org.deeplearning4j.nn.layers.convolution.subsampling

Interface SubsamplingHelper

All Known Implementing Classes:

CudnnSubsamplingHelper

public interface SubsamplingHelper

Helper for the subsampling layer.

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	activate(org.nd4j.linalg.api.ndarray.INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode)
Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray>	backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, 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 epsilon,
                                                                     int[] kernel,
                                                                     int[] strides,
                                                                     int[] pad,
                                                                     PoolingType poolingType,
                                                                     ConvolutionMode convolutionMode)

activate

org.nd4j.linalg.api.ndarray.INDArray activate(org.nd4j.linalg.api.ndarray.INDArray input,
                                              boolean training,
                                              int[] kernel,
                                              int[] strides,
                                              int[] pad,
                                              PoolingType poolingType,
                                              ConvolutionMode convolutionMode)