torch.Tensor.copy_

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

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

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

参数

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

• non_blocking (bool, optional) – 如果为 True 并且此复制是在 CPU 和 GPU 之间进行的，则复制可能相对于主机异步进行。对于其他情况，此参数没有影响。默认为 False