mindspore.Tensor.to

Tensor.to(dtype) → Tensor

Returns a tensor with the new specified data type.

Note

When converting complex numbers to boolean type, the imaginary part of the complex number is not taken into account. As long as the real part is non-zero, it returns True; otherwise, it returns False.

Parameters

dtype (dtype.Number) – The valid data type of the output tensor. Only constant value is allowed.

Returns

Tensor, the data type of the tensor is dtype.

Raises

TypeError – If dtype is not a Number.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> input_np = np.random.randn(2, 3, 4, 5).astype(np.float32)
>>> input = Tensor(input_np)
>>> dtype = mindspore.int32
>>> output = input.to(dtype)
>>> print(output.dtype)
Int32
>>> print(output.shape)
(2, 3, 4, 5)