tf.to_bfloat16

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

将张量强制转换为bfloat16类型。

参数:

• x: 张量或稀疏张量。
• name: 操作的名称(可选)。

返回值:

• 与x形状相同的张量或稀疏张量，类型为bfloat16。

异常:

• TypeError: If x cannot be cast to the bfloat16.

原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/to_bfloat16?hl=en