public interface SubsamplingHelper extends LayerHelper
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() |
helperMemoryUse
boolean checkSupported()
Pair<Gradient,INDArray> backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
INDArray activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
Copyright © 2018. All rights reserved.