mindspore.Tensor.to
- mindspore.Tensor.to(dtype)[源代码]
执行Tensor类型的转换。
说明
如果 self 张量已经具有所需的 mindspore.dtype,则直接返回 self。 否则,将返回一个具有目标 mindspore.dtype 的 self 的副本。
当将复数转换为布尔类型时,复数的虚部不被考虑。只要实部非零,它就返回True;否则,返回False。
- 参数:
dtype (dtype.Number) - 输出Tensor的有效数据类型,只允许常量值。
- 返回:
Tensor,其数据类型为 dtype。
- 异常:
TypeError -如果 dtype 不是数值类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> import mindspore >>> from mindspore import Tensor >>> input_np = np.random.randn(2, 3, 4, 5).astype(np.float32) >>> input_x = Tensor(input_np) >>> dtype = mindspore.int32 >>> output = input_x.to(dtype) >>> print(output.dtype) Int32