[参数完全一致 ]torch.Tensor.masked_select

torch.Tensor.masked_select

torch.Tensor.masked_select(mask)

paddle.Tensor.masked_select

paddle.Tensor.masked_select(mask, name=None)

两者功能一致,参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
mask mask 表示用于索引的二进制掩码的 Tensor 。