megengine.functional.copy

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

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

参数
  • inp – 输入张量。

  • device – 目标设备。

实际案例

>>> import numpy as np
>>> import platform
>>> from megengine.device import get_device_count
>>> 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]