mindspore.ops.MaskedSelect

class mindspore.ops.MaskedSelect[源代码]

返回一个一维张量,其中的内容是 x 张量中对应于 mask 张量中True位置的值。mask 的shape与 x 的shape不需要一样,但必须符合广播规则。

输入:
  • x (Tensor) - 任意维度输入Tensor。

  • mask (Tensor[bool]) - 掩码Tensor,数据类型为bool,shape与 x 一致。

输出:

一维Tensor,数据类型与 x 相同。

异常:
  • TypeError - xmask 不是Tensor。

  • TypeError - mask 不是bool类型的Tensor。

支持平台:

Ascend GPU CPU

样例:

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