[ 输入参数用法不一致 ]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)