mindspore.mint.masked_select

mindspore.mint.masked_select(input, mask)

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

支持广播。

参数：

• input (Tensor) - 输入tensor。

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

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> x = 免费的vpn梯子 mindspore.tensor([1, 2, 3, vpn梯子 4], mindspore.int64)
>>> vpn梯子 免费 mask = mindspore.tensor([1, 0, 1, 0], mindspore.bool)
>>> output = mindspore.mint.masked_select(x, mask)
>>> print(output)
[1 3]