torch.Tensor.copy_

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

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

张量src必须能够与张量self进行广播操作。它可能具有不同的数据类型或位于不同的设备上。

参数

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

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