masked_select

paddle. masked_select ( x, mask, name=None ) [源代码]

返回一个 1-D 的 Tensor,Tensor 的值是根据 mask 对输入 x 进行选择的,mask 的数据类型是 bool。

参数

  • x (Tensor) - 输入 Tensor,数据类型为 float32,float64,int32 或者 int64。

  • mask (Tensor) - 用于索引的二进制掩码的 Tensor,数据类型为 bool。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

返回一个根据 mask 选择的的 Tensor。

代码示例

import paddle

x = paddle.to_tensor([[1.0, 2.0, 3.0, 4.0],
                      [5.0, 6.0, 7.0, 8.0],
                      [9.0, 10.0, 11.0, 12.0]])
mask = paddle.to_tensor([[True, False, False, False],
                         [True, True, False, False],
                         [True, False, False, False]])
out = paddle.masked_select(x, mask)
#[1.0 5.0 6.0 9.0]

使用本API的教程文档