mindspore.Tensor.to
- mindspore.Tensor.to(dtype)
执行Tensor类型的转换。
- 参数:
dtype (Number) - 输出Tensor的有效数据类型,只允许常量值。
- 返回:
Tensor,其数据类型为 dtype。
- 异常:
TypeError -如果 dtype 不是数值类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input_np = np.random.randn(2, 3, 4, 5).astype(np.float32) >>> input = Tensor(input_np) >>> dtype = mindspore.int32 >>> output = ops.to(input, dtype) >>> print(output.dtype) Int32 >>> print(output.shape) (2, 3, 4, 5)