torch.Tensor.copy_

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

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

src 张量必须与 self 张量是可广播的。它可以是不同的数据类型或驻留在不同的设备上。

参数

• src (Tensor) – 要复制的源张量

• non_blocking (bool) – 如果为 True 且此复制发生在 CPU 和 GPU 之间，则复制可能会相对于主机异步发生。对于其他情况，此参数无效。