masked_select

dragon.masked_select(
  inputs,
  **kwargs
)[source]

Select the elements of input where mask is 1.

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

dragon.Tensor – The output tensor.