org.deeplearning4j.util

Class MaskedReductionUtil

java.lang.Object

org.deeplearning4j.util.MaskedReductionUtil

public class MaskedReductionUtil
extends Object

Method Summary

Modifier and Type	Method and Description
static INDArray	maskedPoolingConvolution(PoolingType poolingType, INDArray toReduce, INDArray mask, int pnorm, DataType dataType)
static INDArray	maskedPoolingEpsilonCnn(PoolingType poolingType, INDArray input, INDArray mask, INDArray epsilon2d, int pnorm, DataType dataType)
static INDArray	maskedPoolingEpsilonTimeSeries(PoolingType poolingType, INDArray input, INDArray mask, INDArray epsilon2d, int pnorm)
static INDArray	maskedPoolingTimeSeries(PoolingType poolingType, INDArray toReduce, INDArray mask, int pnorm, DataType dataType)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

maskedPoolingTimeSeries

public static INDArray maskedPoolingTimeSeries(PoolingType poolingType,
                                               INDArray toReduce,
                                               INDArray mask,
                                               int pnorm,
                                               DataType dataType)

maskedPoolingEpsilonTimeSeries

public static INDArray maskedPoolingEpsilonTimeSeries(PoolingType poolingType,
                                                      INDArray input,
                                                      INDArray mask,
                                                      INDArray epsilon2d,
                                                      int pnorm)

maskedPoolingConvolution

public static INDArray maskedPoolingConvolution(PoolingType poolingType,
                                                INDArray toReduce,
                                                INDArray mask,
                                                int pnorm,
                                                DataType dataType)

maskedPoolingEpsilonCnn

public static INDArray maskedPoolingEpsilonCnn(PoolingType poolingType,
                                               INDArray input,
                                               INDArray mask,
                                               INDArray epsilon2d,
                                               int pnorm,
                                               DataType dataType)