mindspore.Tensor.half

查看源文件
mindspore.Tensor.half()[源代码]

将输入Tensor转换为 float16 类型。

返回:

Tensor,其数据类型为 float16

支持平台:

Ascend GPU CPU

样例:

>>> 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