mindspore.Tensor.bfloat16

mindspore.Tensor.bfloat16()

将输入Tensor转换为 bfloat16 类型。

返回：

Tensor，其数据类型为 bfloat16。

支持平台：

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.bfloat16()
>>> print(output.dtype)
BFloat16