mindspore.ops.MaskedSelect
- class mindspore.ops.MaskedSelect[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.
- Inputs:
x (Tensor) - Input Tensor of any dimension.
mask (Tensor[bool]) - Boolean mask Tensor, has the same shape as x.
- Outputs:
A 1-D Tensor, with the same type as x.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int32) >>> mask = Tensor(np.array([1, 0, 1, 0]), mindspore.bool_) >>> output = ops.MaskedSelect()(x, mask) >>> print(output) [1 3] >>> x = Tensor(2.1, mindspore.float32) >>> mask = Tensor(True, mindspore.bool_) >>> output = ops.MaskedSelect()(x, mask) >>> print(output) [2.1]