mindspore.ops.masked_select

mindspore.ops.masked_select(input, mask)

返回一个一维tensor，它根据掩码对输入tensor进行索引。

支持广播。

参数：

• input (Tensor) - 输入tensor。

• mask (Tensor[bool]) - 输入掩码。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

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