テンソルフロー::作戦:: ResourceScatterNdAdd

#include <state_ops.h>

Variable内の個々の値またはスライスにスパース加算を適用します。

まとめ

refはランクPTensorindicesはランクQTensorです。

indicesrefへのインデックスを含む整数テンソルでなければなりません。これは形状[d_0, ..., d_{Q-2}, K] 0 < K <= Pである必要があります。

indicesの最も内側の次元 (長さK ) は、 refK番目の次元に沿った要素 ( K = Pの場合) またはスライス ( K < Pの場合) へのインデックスに対応します。

updates 、次の形状を持つランクQ-1+PKTensorです。

[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 の場合、割り当てはロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。

戻り値:

コンストラクターとデストラクター

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
)