torch.Tensor.copy_

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

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

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

Parameters

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

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