org.deeplearning4j.util

Class MaskedReductionUtil

java.lang.Object

org.deeplearning4j.util.MaskedReductionUtil

public class MaskedReductionUtil
extends java.lang.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.

Constructor Summary

Constructors

Constructor and Description
MaskedReductionUtil()

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	maskedPoolingConvolution(PoolingType poolingType, org.nd4j.linalg.api.ndarray.INDArray toReduce, org.nd4j.linalg.api.ndarray.INDArray mask, boolean alongHeight, int pnorm)
static org.nd4j.linalg.api.ndarray.INDArray	maskedPoolingEpsilonCnn(PoolingType poolingType, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray epsilon2d, boolean alongHeight, int pnorm)
static org.nd4j.linalg.api.ndarray.INDArray	maskedPoolingEpsilonTimeSeries(PoolingType poolingType, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray mask, org.nd4j.linalg.api.ndarray.INDArray epsilon2d, int pnorm)
static org.nd4j.linalg.api.ndarray.INDArray	maskedPoolingTimeSeries(PoolingType poolingType, org.nd4j.linalg.api.ndarray.INDArray toReduce, org.nd4j.linalg.api.ndarray.INDArray mask, int pnorm)

Methods inherited from class java.lang.Object

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

Constructor Detail

MaskedReductionUtil

public MaskedReductionUtil()

Method Detail

maskedPoolingTimeSeries

public static org.nd4j.linalg.api.ndarray.INDArray maskedPoolingTimeSeries(PoolingType poolingType,
                                                                           org.nd4j.linalg.api.ndarray.INDArray toReduce,
                                                                           org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                           int pnorm)

maskedPoolingEpsilonTimeSeries

public static org.nd4j.linalg.api.ndarray.INDArray maskedPoolingEpsilonTimeSeries(PoolingType poolingType,
                                                                                  org.nd4j.linalg.api.ndarray.INDArray input,
                                                                                  org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                                  org.nd4j.linalg.api.ndarray.INDArray epsilon2d,
                                                                                  int pnorm)

maskedPoolingConvolution

public static org.nd4j.linalg.api.ndarray.INDArray maskedPoolingConvolution(PoolingType poolingType,
                                                                            org.nd4j.linalg.api.ndarray.INDArray toReduce,
                                                                            org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                            boolean alongHeight,
                                                                            int pnorm)

maskedPoolingEpsilonCnn

public static org.nd4j.linalg.api.ndarray.INDArray maskedPoolingEpsilonCnn(PoolingType poolingType,
                                                                           org.nd4j.linalg.api.ndarray.INDArray input,
                                                                           org.nd4j.linalg.api.ndarray.INDArray mask,
                                                                           org.nd4j.linalg.api.ndarray.INDArray epsilon2d,
                                                                           boolean alongHeight,
                                                                           int pnorm)