torch.Tensor.copy_

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

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

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

参数

• src (Tensor) – 源张量，用于复制

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