TensorFlow函数：tf.to_bfloat16

tf.to_bfloat16函数

tf.to_bfloat16(
    x,
    name='ToBFloat16'
)

定义在：tensorflow/python/ops/math_ops.py.

参见指南：张量转换

将张量转换为 bfloat16 类型.

函数参数：

• x：一个 Tensor 或 SparseTensor.
• name：操作的名称(可选).

函数返回值：

tf.to_bfloat16函数返回一个 Tensor 或 SparseTensor 与 x(类型为 bfloat16)具有相同的形状.

可能引发的异常：

• TypeError：如果 x 不能转换为 bfloat16.