aliran tensor:: operasi:: ResourceScatterNdSub

#include <state_ops.h>

Menerapkan pengurangan renggang pada nilai atau irisan individual dalam Variabel .

Ringkasan

ref adalah Tensor dengan peringkat P dan indices adalah Tensor dengan peringkat Q .

indices harus berupa tensor bilangan bulat, berisi indeks ke dalam ref . Itu harus berbentuk [d_0, ..., d_{Q-2}, K] di mana 0 < K <= P .

Dimensi terdalam dari indices (dengan panjang K ) sesuai dengan indeks menjadi elemen (jika K = P ) atau irisan (jika K < P ) sepanjang K dimensi ke- ref .

updates adalah Tensor peringkat Q-1+PK dengan bentuk:

[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]

Misalnya, kita ingin mengurangi 4 elemen tersebar dari tensor peringkat-1 dengan 8 elemen. Dengan Python, pengurangan itu akan terlihat seperti ini:

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])
sub = tf.scatter_nd_sub(ref, indices, updates)
with tf.Session() as sess:
  print sess.run(sub)

Pembaruan yang dihasilkan untuk ref akan terlihat seperti ini:

[1, -9, 3, -6, -4, 6, 7, -4]

Lihat tf.scatter_nd untuk detail selengkapnya tentang cara memperbarui irisan.

Argumen:

  • ruang lingkup: Objek Lingkup
  • ref: Pegangan sumber daya. Harus dari VarHandleOp.
  • indeks: Tensor . Harus berupa salah satu dari jenis berikut: int32, int64. Tensor indeks menjadi ref.
  • pembaruan: Tensor . Harus memiliki tipe yang sama dengan ref. Tensor nilai untuk ditambahkan ke referensi.

Atribut opsional (lihat Attrs ):

  • use_locking: Bool opsional. Defaultnya adalah Benar. Jika Benar, penugasan akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.

Pengembalian:

Konstruktor dan Destruktor

ResourceScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates)
ResourceScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ResourceScatterNdSub::Attrs & attrs)

Atribut publik

operation

Fungsi publik

operator::tensorflow::Operation () const

Fungsi statis publik

UseLocking (bool x)

Struktur

tensorflow:: ops:: ResourceScatterNdSub:: Attrs

Penyetel atribut opsional untuk ResourceScatterNdSub .

Atribut publik

operasi

Operation operation

Fungsi publik

ResourceScatterNdSub

 ResourceScatterNdSub(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates
)

ResourceScatterNdSub

 ResourceScatterNdSub(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates,
  const ResourceScatterNdSub::Attrs & attrs
)

operator::tensorflow::Operasi

 operator::tensorflow::Operation() const 

Fungsi statis publik

Gunakan Penguncian

Attrs UseLocking(
  bool x
)