mindspore.Tensor.half

View Source On Gitee
Tensor.half()[source]

Converts input tensor dtype to float16.

Returns

Tensor, converted to the float16 dtype.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> input_x = Tensor(np.ones([2,2]), mindspore.int32)
>>> output = input_x.half()
>>> print(output.dtype)
Float16