mindspore.Tensor.to

View Source On Gitee
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