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

torch.Tensor.sparse_mask

torch.Tensor.sparse_mask(mask)

paddle.sparse.mask_as

paddle.sparse.mask_as(x, mask, name=None)

两者功能一致,但调用方式不同,torch 通过 Tensor 类方法调用,而 paddle 是直接调用函数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
- x 输入的 DenseTensor。
mask mask 掩码逻辑的 mask,参数完全一致。

转写示例

# torch 调用 Tensor 类方法
x.sparse_mask(mask)

# paddle 直接调用函数
paddle.sparse.mask_as(x, mask)