（beta）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。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas A3 训练系列产品
• Atlas 推理系列产品

调用示例

1
2
3
4

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