mindspore.ops.MaskedSelect

class mindspore.ops.MaskedSelect(*args, **kwargs)[source]

Returns a new 1-D Tensor which indexes the input tensor according to the boolean mask. The shapes of the mask tensor and the input 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 is not a Tensor.

Supported Platforms:: CPU

Examples

>>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int64)
>>> mask = Tensor(np.array([1, 0, 1, 0]), mindspore.bool_)
>>> output = ops.MaskedSelect()(x, mask)
>>> print(output)
[1 3]