CrossReplicaSum

パブリック最終クラスCrossReplicaSum

複製されたTPUインスタンス間で入力を合計するための操作。

各インスタンスは独自の入力を提供します。

たとえば、8つのTPUインスタンスがあるとします： `[A、B、C、D、E、F、G、H]`。 group_assignment = `[[0,2,4,6]、[1,3,5,7]]`を渡すと、 `A、C、E、G`がグループ0として設定され、` B、D、F、H`が設定されます。したがって、出力は次のようになります。 `[A + C + E + G、B + D + F + H、A + C + E + G、B + D + F + H、A + C + E + G、B + D + F + H、A + C + E + G、B + D + F + H] `。

パブリックメソッド

出力<T>
asOutput （）
テンソルのシンボリックハンドルを返します。
静的<Tが数拡張> CrossReplicaSum <T>
作成（スコープ範囲、オペランド<T>入力、オペランド<整数> groupAssignment）
新しいCrossReplicaSum操作をラップするクラスを作成するファクトリメソッド。
出力<T>
出力（）
すべての分散入力の合計。

継承されたメソッド

パブリックメソッド

公共の出力<T> asOutput（）

テンソルのシンボリックハンドルを返します。

TensorFlow操作への入力は、別のTensorFlow操作の出力です。このメソッドは、入力の計算を表すシンボリックハンドルを取得するために使用されます。

パブリック静的CrossReplicaSum <T>を作成（スコープ範囲、オペランド<T>入力、オペランド<整数> groupAssignment）

新しいCrossReplicaSum操作をラップするクラスを作成するファクトリメソッド。

パラメーター
範囲現在のスコープ
入力合計へのローカル入力。
groupAssignment形状が[num_groups、num_replicas_per_group]のint32テンソル。 `group_assignment [i]`は、i番目のサブグループのレプリカIDを表します。
戻り値
  • CrossReplicaSumの新しいインスタンス

公共の出力<T>出力（）

すべての分散入力の合計。