[ 输入参数用法不一致 ]torch.Tensor.where

torch.Tensor.where

torch.Tensor.where(condition, other)

paddle.where

paddle.where(condition, x=None, y=None, name=None)

Pytorch 为 Tensor 类方法,Paddle 为普通函数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
condition condition 判断条件。
self x 当 condition 为 true 时,选择的元素,调用 torch.Tensor 类方法的 self Tensor 传入。
other y 当 condition 为 false 时,选择的元素,仅参数名不一致。

转写示例

# torch 写法
a = torch.tensor([0, 1, 2])
b = torch.tensor([2, 3, 0])
c = a.where(a > 0, b)

# paddle 写法
paddle.where(a > 0, a, b)