aliran tensor:: operasi:: Tambahkan Sebar
#include <state_ops.h>
Menambahkan pembaruan jarang ke referensi variabel.
Ringkasan
Operasi ini menghitung
# Scalar indices ref[indices, ...] += updates[...] # Vector indices (for each i) ref[indices[i], ...] += updates[i, ...] # High rank indices (for each i, ..., j) ref[indices[i, ..., j], ...] += updates[i, ..., j, ...]
Operasi ini mengeluarkan ref
setelah pembaruan selesai. Hal ini mempermudah operasi berantai yang perlu menggunakan nilai reset.
Entri duplikat ditangani dengan benar: jika beberapa indices
merujuk pada lokasi yang sama, kontribusinya akan bertambah.
Memerlukan updates.shape = indices.shape + ref.shape[1:]
atau updates.shape = []
.
Argumen:
- ruang lingkup: Objek Lingkup
- ref: Harus dari node
Variable
. - indices: Tensor indeks ke dalam dimensi pertama
ref
. - update: Tensor nilai yang diperbarui untuk ditambahkan ke
ref
.
Atribut opsional (lihat Attrs
):
- use_locking: Jika Benar, penambahan akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.
Pengembalian:
-
Output
: = Sama sepertiref
. Dikembalikan untuk memudahkan operasi yang ingin menggunakan nilai yang diperbarui setelah pembaruan selesai.
Konstruktor dan Destruktor | |
---|---|
ScatterAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterAdd::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
output_ref |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
UseLocking (bool x) |
Struktur | |
---|---|
tensorflow:: ops:: ScatterAdd:: Attrs | Penyetel atribut opsional untuk ScatterAdd . |
Atribut publik
operasi
Operation operation
keluaran_ref
::tensorflow::Output output_ref
Fungsi publik
Tambahkan Sebar
ScatterAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
Tambahkan Sebar
ScatterAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterAdd::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
Gunakan Penguncian
Attrs UseLocking( bool x )