mindspore.ops.masked_select
- mindspore.ops.masked_select(input, mask)[source]
Returns a new 1-D Tensor which indexes the x tensor according to the boolean mask. The shapes of the mask tensor and the x tensor don't need to match, but they must be broadcastable.
- Parameters
- Returns
A 1-D Tensor, with the same type as input.
- Raises
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> import mindspore >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int64) >>> mask = Tensor(np.array([1, 0, 1, 0]), mindspore.bool_) >>> output = ops.masked_select(x, mask) >>> print(output) [1 3]