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')