警告:このAPIは非推奨であり、置き換えが安定した後 TensorFlowの将来のバージョンで削除される予定です。

このページは Cloud Translation API によって翻訳されました。
Switch to English

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 ()
テンソルのシンボリックハンドルを返します。
static <T extends Number> CrossReplicaSum <T>
createスコープスコープ、オペランド<T>入力、オペランド<整数> groupAssignment)
新しいCrossReplicaSum操作をラップするクラスを作成するファクトリメソッド。
出力<T>
出力()
すべての分散入力の合計。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

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

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

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

public Output <T> output ()

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