[ 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)