boolean_mask¶

dragon.boolean_mask(
  inputs,
  **kwargs
)[source]¶

Return the elements of input where mask is true.

Parameters:

• inputs (Sequence[dragon.Tensor]) – The input and mask tensor.

Returns:

dragon.Tensor – The output tensor.