CrossReplicaSum

パブリック最終クラスCrossReplicaSum

レプリケートされた TPU インスタンス全体の入力を合計する Op。

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

たとえば、「[A、B、C、D、E、F、G、H]」という 8 つの TPU インスタンスがあるとします。 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 ()
テンソルのシンボリック ハンドルを返します。
static <T extends Number> CrossReplicaSum <T>
create (スコープscope、オペランド<T>入力、オペランド<Integer>groupAssignment)
新しい CrossReplicaSum オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
出力()
すべての分散入力の合計。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CrossReplicaSum <T> create (スコープscope、オペランド<T>入力、オペランド<Integer> groupAssignment)

新しい CrossReplicaSum オペレーションをラップするクラスを作成するためのファクトリ メソッド。

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

public出力<T>出力()

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