torch_npu.copy_memory_

torch_npu.copy_memory_(dst, src, non_blocking=False) -> Tensor

从src拷贝元素到self张量，并返回self。

• 参数解释：
  • dst (Tensor) - 拷贝源张量。
  • src (Tensor) - 返回张量所需数据类型。
  • non_blocking (Bool,默认值为False) - 如果设置为True且此拷贝位于CPU和NPU之间，则拷贝可能相对于主机异步发生。在其他情况下，此参数没有效果。
• 约束条件：

  copy_memory_仅支持NPU张量。copy_memory_的输入张量应具有相同的dtype和设备index。

• 示例：

  >>> a=torch.IntTensor([0,  0, -1]).npu()
  >>> b=torch.IntTensor([1, 1, 1]).npu()
  >>> a.copy_memory_(b)
  tensor([1, 1, 1], device='npu:0', dtype=torch.int32)