Python tf.keras.backend.set_floatx用法及代码示例

设置默认浮点类型。

用法

tf.keras.backend.set_floatx(
    value
)

注意：不建议将其设置为 float16 进行训练，因为这可能会导致数值稳定性问题。相反，混合精度，它使用 float16 和 float32 的混合，可以通过调用来使用tf.keras.mixed_precision.set_global_policy('mixed_float16').见混合精度导轨详情。

例子：

tf.keras.backend.floatx()
'float32'
tf.keras.backend.set_floatx('float64')
tf.keras.backend.floatx()
'float64'
tf.keras.backend.set_floatx('float32')