aliran tensor:: operasi:: ScatterMax
#include <state_ops.h>
Mengurangi pembaruan yang jarang menjadi referensi variabel menggunakan operasi max
.
Ringkasan
Operasi ini menghitung
# Scalar indices ref[indices, ...] = max(ref[indices, ...], updates[...]) # Vector indices (for each i) ref[indices[i], ...] = max(ref[indices[i], ...], updates[i, ...]) # High rank indices (for each i, ..., j) ref[indices[i, ..., j], ...] = max(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 digabungkan.
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 direduksi menjadi
ref
.
Atribut opsional (lihat Attrs
):
- use_locking: Jika Benar, pembaruan akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan.
Pengembalian:
-
Output
: = Sama denganref
. Dikembalikan untuk memudahkan operasi yang ingin menggunakan nilai yang diperbarui setelah pembaruan selesai.
Konstruktor dan Destruktor | |
---|---|
ScatterMax (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterMax (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterMax::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:: ScatterMax:: Attrs | Penyetel atribut opsional untuk ScatterMax . |
Atribut publik
operasi
Operation operation
keluaran_ref
::tensorflow::Output output_ref
Fungsi publik
ScatterMax
ScatterMax( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ScatterMax
ScatterMax( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterMax::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 )