[ torch 参数更多 ] torch.Tensor.copy_

torch.Tensor.copy_

torch.Tensor.copy_(src, non_blocking=False)

paddle.assign

paddle.assign(x, output=None)

两者功能类似,torch 参数多,具体如下:

参数差异

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | src | x | 待复制的 tensor | | non_blocking | - | 用于控制 cpu 和 gpu 数据的异步复制,转写无需考虑该参数 |

转写示例

# torch 写法
z.copy_(x)

# paddle 写法
paddle.assign(x, output=z)