[ 仅 API 调用方式不一致 ]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)

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
out = x.sparse_mask(coo)

# Paddle 写法
out = paddle.sparse.mask_as(x, coo)