mindspore.Tensor.astype

mindspore.Tensor.astype(dtype, copy=True)[源代码]

将Tensor转为指定数据类型，可指定是否返回副本。

参数：

  • dtype (Union[mindspore.dtype, numpy.dtype, str]) - 指定的Tensor数据类型，可以是: mindspore.dtype.float32 , numpy.float32 或 float32 的格式。默认值：mindspore.dtype.float32 。

  • copy (bool, 可选) - 默认情况下，astype返回新拷贝的Tensor。如果该参数设为False，则返回输入Tensor而不是副本。默认值：True。

返回：

Tensor，指定数据类型的Tensor。

异常：

  • TypeError - 指定了无法解析的类型。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,2,1), dtype=np.float32))
>>> x = x.astype("int32")
>>> print(x.dtype)
Int32