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

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


將 N 個輸入連接到 N-way 複製的 TPU 計算。

用法

tf.raw_ops.TPUReplicatedInput(
    inputs, is_mirrored_variable=False, index=-1, is_packed=False, name=None
)

參數

  • inputs 至少 1 個具有相同類型的 Tensor 對象的列表。
  • is_mirrored_variable 可選的 bool 。默認為 False 。
  • index 可選的 int 。默認為 -1 。
  • is_packed 可選的 bool 。默認為 False 。
  • name 操作的名稱(可選)。

返回

  • 一個Tensor。具有與 inputs 相同的類型。

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

例如：

%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)

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

相關用法


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