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.

Raises

TypeError – If x or mask is not a Tensor.
TypeError – If dtype of mask is not bool.

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]