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) - The shape of tensor is \((x_1, x_2, ..., x_R)\).
mask (Tensor[bool]) - The shape of tensor is \((x_1, x_2, ..., x_R)\).
- Outputs:
A 1-D Tensor, with the same type as x.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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]