mindspore.Tensor.to
- mindspore.Tensor.to(dtype)[源代码]
执行Tensor类型的转换。
- 参数:
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