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

Python tf.tpu.experimental.DeviceAssignment用法及代碼示例

從計算中的邏輯核心映射到物理 TPU 拓撲。

用法

tf.tpu.experimental.DeviceAssignment(
    topology:tf.tpu.experimental.Topology,
    core_assignment:np.ndarray
)

參數

  • topology 說明物理 TPU 拓撲的 Topology 對象。
  • core_assignment 邏輯到物理核心映射，表示為 rank 3 numpy 數組。有關詳細信息，請參閱core_assignment 屬性的說明。

拋出

  • ValueError 如果 topology 不是 Topology 對象。
  • ValueError 如果 core_assignment 不是 3 級 numpy 數組。

屬性

  • core_assignment 邏輯到物理核心映射。
  • num_cores_per_replica 每個副本的核心數。
  • num_replicas 計算的副本數。
  • topology 說明 TPU 拓撲的Topology。

更喜歡使用 DeviceAssignment.build() 幫助器來構造 DeviceAssignment ；如果靈活性不如直接構建DeviceAssignment，那麽它會更容易。

相關用法


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