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

torch.Tensor.copy_¶

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

         

          paddle.assign(x, output=None)

         

两者功能类似，torch 参数多，具体如下：

PyTorch	PaddlePaddle	备注
src	x	待复制的 tensor，仅参数名不一致。
non_blocking	-	用于控制 cpu 和 gpu 数据的异步复制。Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。
-	output	输出 Tensor，PyTorch 无此参数，Paddle 需将其设置为调用 copy_类方法的 Tensor 。

          # torch 写法
y.copy_(x)

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