Package org.nd4j.linalg.indexing.masking

Class Masking

java.lang.Object

org.nd4j.linalg.indexing.masking.Masking

public class Masking
extends Object

Constructor Summary

Constructors

Constructor	Description
Masking()

Method Summary

Modifier and Type	Method	Description
static SDVariable	applyMask(SameDiff ret, SDVariable input, SDVariable mask, int axis)
static INDArray	applyMask(INDArray input, INDArray mask, int axis)	Applies a boolean mask to the given input.

Methods inherited from class java.lang.Object

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

Constructor Detail

Masking

public Masking()

Method Detail

applyMask

public static SDVariable applyMask(SameDiff ret,
                                   SDVariable input,
                                   SDVariable mask,
                                   int axis)

applyMask

public static INDArray applyMask(INDArray input,
                                 INDArray mask,
                                 int axis)

Applies a boolean mask to the given input. This is equivalent to tensorflow's boolean_mask

Parameters:

input - the input to mask

mask - the target mask to apply

axis - the axis to apply along

Returns: