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


Python tf.distribute.cluster_resolver.TPUClusterResolver.get_tpu_system_metadata用法及代碼示例


用法

get_tpu_system_metadata()

返回 TPU 係統的元數據。

用戶可以調用此方法來獲取 TPU 係統的一些信息,例如內核總數、TPU 工作器數量和設備。例如:

resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
tpu_system_metadata = resolver.get_tpu_system_metadata()
num_hosts = tpu_system_metadata.num_hosts

相關用法


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