mindspore.Tensor.to
- Tensor.to(dtype)[source]
Performs tensor dtype conversion.
- Parameters
dtype (Number) – The valid data type of the output tensor. Only constant value is allowed.
- Returns
Tensor, converted to the specified dtype.
- Raises
TypeError – If dtype is not a Number.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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