当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。