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 - x 或 mask 不是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]