tf.keras.backend.set_floatx

Устанавливает тип плавающей точки по умолчанию.

tf.keras.backend.set_floatx(
    value
)

Примечание: Не рекомендуется устанавливать это значение в float16 для обучения, так как это, вероятно, приведет к проблемам с числовой устойчивостью. Вместо этого можно использовать смешанную точность, которая использует комбинацию float16 и float32, вызвав tf.keras.mixed_precision.experimental.set_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')