テンソルフロー::作戦:: ResourceScatterNdAdd
#include <state_ops.h>
Variable内の個々の値またはスライスにスパース加算を適用します。
まとめ
ref
はランクP
のTensor
、 indices
はランクQ
のTensor
です。
indices
、 ref
へのインデックスを含む整数テンソルでなければなりません。これは形状[d_0, ..., d_{Q-2}, K]
0 < K <= P
である必要があります。
indices
の最も内側の次元 (長さK
) は、 ref
のK
番目の次元に沿った要素 ( K = P
の場合) またはスライス ( K < P
の場合) へのインデックスに対応します。
updates
、次の形状を持つランクQ-1+PK
のTensor
です。
[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]
たとえば、8 要素のランク 1 テンソルに 4 つの散乱要素を追加したいとします。 Python では、この追加は次のようになります。
ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8], use_resource=True) indices = tf.constant([[4], [3], [1], [7]]) updates = tf.constant([9, 10, 11, 12]) add = tf.scatter_nd_add(ref, indices, updates) with tf.Session() as sess: print sess.run(add)
ref への結果の更新は次のようになります。
[1, 13, 3, 14, 14, 6, 7, 20]
スライスを更新する方法の詳細については、 tf.scatter_nd
参照してください。
引数:
- スコープ:スコープオブジェクト
- 参照: リソースハンドル。 VarHandleOp からのものである必要があります。
- インデックス: Tensor 。 int32、int64 のいずれかの型である必要があります。 ref へのインデックスのテンソル。
- 更新: Tensor 。 ref と同じ型でなければなりません。 ref に追加する値のテンソル。
オプションの属性 ( Attrs
を参照):
- use_locking: オプションのブール値。デフォルトは True です。 True の場合、割り当てはロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。
戻り値:
- 作成された
Operation
コンストラクターとデストラクター | |
---|---|
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs) |
パブリック属性 | |
---|---|
operation |
公共機能 | |
---|---|
operator::tensorflow::Operation () const |
パブリック静的関数 | |
---|---|
UseLocking (bool x) |
構造体 | |
---|---|
tensorflow:: ops:: ResourceScatterNdAdd:: Attrs | ResourceScatterNdAddのオプションの属性セッター。 |
パブリック属性
手術
Operation operation
公共機能
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs )
演算子::tensorflow::オペレーション
operator::tensorflow::Operation() const
パブリック静的関数
ロックを使用する
Attrs UseLocking( bool x )