変数参照にスパース更新を追加します。
この操作で計算されるのは、
# スカラーインデックス ref[indices, ...] += 更新[...]
# ベクトルインデックス (各 i) ref[indices[i], ...] += updated[i, ...]
# 高ランクのインデックス (各 i, ..., j) ref[indices[i, ..., j], ...] += update[i, ..., j, ...]
この操作は、更新が完了した後に `ref` を出力します。これにより、リセット値を使用する必要がある操作をチェーンすることが容易になります。
重複エントリは正しく処理されます。複数の「インデックス」が同じ場所を参照している場合、それらの寄与が加算されます。
`updates.shape = indices.shape + ref.shape[1:]` または `updates.shape = []` が必要です。
ネストされたクラス
クラス | ScatterAdd.オプション | ScatterAdd のオプションの属性 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T, U extends Number> ScatterAdd <T> | |
出力<T> | 出力参照() = `ref` と同じ。 |
静的ScatterAdd.Options | useLocking (ブール型 useLocking) |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static ScatterAdd <T> create (スコープスコープ、オペランド<T> 参照、オペランド<U> インデックス、オペランド<T> 更新、オプション...オプション)
新しい ScatterAdd オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
参照 | 「Variable」ノードからのものである必要があります。 |
インデックス | `ref` の最初の次元へのインデックスのテンソル。 |
アップデート | `ref` に追加する更新された値のテンソル。 |
オプション | オプションの属性値を持ちます |
戻り値
- ScatterAdd の新しいインスタンス
public static ScatterAdd.Options useLocking (ブール型 useLocking)
パラメーター
使用ロック | True の場合、追加はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 |
---|