[ 参数不⼀致 ]torch.Tensor.where¶
torch.Tensor.where¶
torch.Tensor.where(condition, y)
参数差异¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | condition | - | condition 为判断条件。在 paddle 中 condition 需要转写。| | - | x | 当 condition 为 true 时，选择 x 中元素。| | y | y | 当 condition 为 false 时，选择 y 中元素。|
转写示例¶
# torch 写法
a = torch.tensor([0, 1, 2])
b = torch.tensor([2, 3, 0])
c = a.where(a > 0, b)
# paddle 写法
a = paddle.to_tensor([0, 1, 2])
b = paddle.to_tensor([2, 3, 0])
c = (a > 0).where(a, b)