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]

megengine.functional.full_like

megengine.functional.reshape