[ 输入参数用法不一致 ]torch.Tensor.where¶
torch.Tensor.where¶
torch.Tensor.where(condition, y)
参数映射¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | condition | - | condition 为判断条件。Paddle 无此参数,需要转写。| | - | 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)