mindspore.Tensor.move_to

查看源文件
mindspore.Tensor.move_to(to, blocking=True)[源代码]

同步或异步的方式将Tensor拷贝到目标设备上,默认同步方式。只支持PyNative模式。

参数:
  • to (str) - 字符串类型,取值为 "Ascend""GPU""CPU" 其中之一。

  • blocking (bool) - 同步或者异步拷贝方式,blocking默认为True,即同步拷贝。

返回:

存储在目标设备上的新Tensor,与原Tensor有相同的shape和type。

异常:
  • ValueError - 如果 blocking 不是bool类型。

  • ValueError - 如果 to 不是 "Ascend""GPU""CPU" 其中之一。

  • ValueError - 如果执行模式不是PyNative模式。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore import Tensor
>>> x = ms.Tensor([1, 2, 3], ms.int64)
>>> new_tensor = x.move_to("CPU")