megengine.functional.copy

copy(inp, device=None)

Copies tensor to another device.

Parameters

• inp – input tensor.

• device – destination device.

Examples

>>> import numpy as np
>>> x = Tensor([1, 2, 3], np.int32)

>>> F.copy(x, 'cpu1')
Tensor([1 2 3], dtype=int32, device=cpu1:0)

>>> F.copy(x, 'xpu0')
Tensor([1 2 3], dtype=int32, device=xpu0:0)

previous

megengine.functional.polar

next

megengine.functional.reshape