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。