mindspore.ops.MaskedSelect

class mindspore.ops.MaskedSelect

使用布尔掩码对输入进行选择得到一个新的一维Tensor。掩码Tensor和输入Tensor的shape不需要匹配，但需支持广播。

输入：

• x (Tensor) - 输入的Tensor，其shape为 \((x_1, x_2, ..., x_R)\) 。

• mask (Tensor[bool]) - 输入的布尔掩码，其shape为 \((x_1, x_2, ..., x_R)\) 。

输出：

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

异常：

• TypeError - x 不是Tensor。

支持平台：

Ascend CPU

样例：

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