mindspore.ops.masked_select

mindspore.ops.masked_select(input, mask)

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

参数：

• input (Tensor) - 它的shape是 \((x_1, x_2, ..., x_R)\)。

• mask (Tensor[bool]) - 它的shape是 \((x_1, x_2, ..., x_R)\)。

返回：

一个一维Tensor，类型与 input 相同。

异常：

• TypeError - input 或 mask 不是Tensor。

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

支持平台：

Ascend GPU CPU

样例：

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