masked_select

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

Select the elements where the given mask is 1.

Parameters:
Returns:

dragon.Tensor – The output tensor.