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

Python tf.raw_ops.TPUReplicatedOutput用法及代碼示例


連接來自 N-way 複製 TPU 計算的 N 個輸出。

用法

tf.raw_ops.TPUReplicatedOutput(
    input, num_replicas, name=None
)

參數

  • input 一個Tensor。
  • num_replicas int 即 >= 1 。
  • name 操作的名稱(可選)。

返回

  • 與 input 具有相同類型的 num_replicas Tensor 對象的列表。

此操作保存來自 tpu.replicate() 計算子圖的複製輸出。每個複製的輸出與輸入具有相同的形狀和類型。

例如：

%computation = "tf.Computation"()
%replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)

上述計算具有兩個副本的複製輸出。

相關用法


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