megengine.functional.copy

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

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

参数
  • inp – 输入张量。

  • device – 目标设备。

例如:

import numpy as np
import platform
from megengine import tensor
from megengine.device import get_device_count
import megengine.functional as F

x = tensor([1, 2, 3], np.int32)
if 1 == get_device_count("gpu"):
    y = F.copy(x, "cpu1")
    print(y.numpy())
else:
    y = F.copy(x, "xpu1")
    print(y.numpy())

输出:

[1 2 3]