mindspore.Tensor.to_
- mindspore.Tensor.to_(device=None, non_blocking=False)[源代码]
mindspore.Tensor.to()的in-place版本,将原始tensor的设备转换成指定的 device 并返回。警告
这是一个实验性API,后续可能修改或删除。
- 参数:
device (str, 可选) - 用于指定输出tensor所在的硬件设备。默认值是
None。non_blocking (bool, 可选) - 数据异步转换。如果是
True,数据类型异步转换。如果是False,数据同步转换。默认值:False。
- 返回:
Tensor,返回被修改后的 self 自身,其所在的设备为 device。
样例:
>>> import mindspore as ms >>> x = ms.Tensor([1, 2, 3, 4]) >>> x.to_(device="Ascend", non_blocking=True) >>> print(x.device) "Ascend:0"