参考 tf.to_bfloat16 - 云+社区 - 腾讯云
tf.to_bfloat16(
x,
name='ToBFloat16'
)
将张量强制转换为bfloat16类型。
参数:
- x: 张量或稀疏张量。
- name: 操作的名称(可选)。
返回值:
- 与x形状相同的张量或稀疏张量,类型为bfloat16。
异常:
TypeError
: Ifx
cannot be cast to thebfloat16
.
原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/to_bfloat16?hl=en