當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.compat.v1.tpu.bfloat16_scope用法及代碼示例


bfloat16 變量的範圍類，以便模型使用自定義 getter。

用法

@tf_contextlib.contextmanager
tf.compat.v1.tpu.bfloat16_scope(
    name:Optional[Text] = None
) -> Generator[tf.compat.v1.variable_scope, None, None]

參數

  • name 用於範圍的名稱。

生成(Yield)

  • 一個可變範圍。

這使得在使用get_variable 時變量可以被讀取為 bfloat16 類型。

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.compat.v1.tpu.bfloat16_scope。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。