org.deeplearning4j.util

Class MaskedReductionUtil

java.lang.Object

org.deeplearning4j.util.MaskedReductionUtil

public class MaskedReductionUtil
extends Object

This is a TEMPORARY class for implementing global pooling with masking. Note that it may be removed in a future release, if and when these approaches are formally implemented as native operations in ND4J. Consequently, this should not be considered part of the public API.

Author:

Alex Black

Method Summary

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

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)

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)

maskedPoolingEpsilonCnn

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