mindspore.ops.MaskedSelect

View Source On Gitee
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.

Raises
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]