Python tf.keras.mixed_precision.global_policy用法及代码示例

返回全局 dtype 策略。

用法

tf.keras.mixed_precision.global_policy()

如果没有将策略传递给层构造函数，则全局策略是用于层的默认 tf.keras.mixed_precision.Policy。如果没有使用 keras.mixed_precision.set_global_policy 设置策略，这将返回从 tf.keras.backend.floatx() 构造的策略(floatx 默认为 float32)。

tf.keras.mixed_precision.global_policy()
<Policy "float32">
tf.keras.layers.Dense(10).dtype_policy  # Defaults to the global policy
<Policy "float32">

如果已使用 tf.compat.v1.disable_v2_behavior() 禁用了 TensorFlow 2 行为，则这将返回一个特殊的 "_infer" 策略，该策略在第一次调用层时从第一个输入的 dtype 推断 dtype。此行为与 TensorFlow 1 中存在的行为相匹配。

有关政策的更多信息，请参阅tf.keras.mixed_precision.Policy。