torch.Tensor.copy_

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

将 src 中的元素复制到 self 张量中，并返回 self 。

src 张量必须与 self 张量可广播。它们可以具有不同的数据类型或位于不同的设备上。

参数:

• 源（Tensor）- 要复制的源张量

• non_blocking（布尔值）- 如果 True 且此复制是在 CPU 和 GPU 之间，则复制可能相对于主机异步进行。对于其他情况，此参数没有效果。