megengine.functional.copy

copy(inp, device=None)[源代码]

把张量复制到另一个设备上。

参数
  • inp – 输入张量。

  • device – 目标设备。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor([1, 2, 3], np.int32)
y = F.copy(x, "xpu1")
print(y.numpy())

输出:

[1 2 3]