org.deeplearning4j.nn.layers.convolution.subsampling

Interface SubsamplingHelper

All Superinterfaces:

LayerHelper

All Known Implementing Classes:

MKLDNNSubsamplingHelper

public interface SubsamplingHelper
extends LayerHelper

Helper for the subsampling layer.

Author:

saudet

Method Summary

Modifier and Type	Method and Description
INDArray	activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>	backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported()

Methods inherited from interface org.deeplearning4j.nn.layers.LayerHelper

helperMemoryUse

Method Detail

checkSupported

boolean checkSupported()

backpropGradient

Pair<Gradient,INDArray> backpropGradient(INDArray input,
                                         INDArray epsilon,
                                         int[] kernel,
                                         int[] strides,
                                         int[] pad,
                                         PoolingType poolingType,
                                         ConvolutionMode convolutionMode,
                                         int[] dilation,
                                         LayerWorkspaceMgr workspaceMgr)

activate

INDArray activate(INDArray input,
                  boolean training,
                  int[] kernel,
                  int[] strides,
                  int[] pad,
                  PoolingType poolingType,
                  ConvolutionMode convolutionMode,
                  int[] dilation,
                  LayerWorkspaceMgr workspaceMgr)