torch.Tensor.copy_

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

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

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

参数

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

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