aliran tensor:: operasi:: SparseSoftmax
#include <sparse_ops.h>
Menerapkan softmax ke ND SparseTensor
batch.
Ringkasan
Input mewakili ND SparseTensor dengan bentuk logis [..., B, C]
(di mana N >= 2
), dan dengan indeks yang diurutkan dalam urutan leksikografis kanonik.
Operasi ini setara dengan menerapkan tf.nn.softmax()
normal ke setiap submatriks logis terdalam dengan bentuk [B, C]
, tetapi dengan ketentuan bahwa elemen nol yang secara implisit tidak berpartisipasi . Secara khusus, algoritma ini setara dengan berikut ini:
(1) Menerapkan tf.nn.softmax()
ke tampilan yang dipadatkan dari setiap submatriks terdalam dengan bentuk [B, C]
, sepanjang dimensi ukuran-C; (2) Menutupi lokasi asli yang secara implisit nol; (3) Menormalkan kembali elemen yang tersisa.
Oleh karena itu, hasil SparseTensor
memiliki indeks dan bentuk bukan nol yang persis sama.
Argumen:
- ruang lingkup: Objek Lingkup
- sp_indices: 2-D. Matriks
NNZ x R
dengan indeks nilai tidak kosong dalam SparseTensor, dalam urutan kanonik. - sp_values: 1-D. Nilai
NNZ
yang tidak kosong sesuai dengansp_indices
. - sp_bentuk: 1-D. Bentuk masukan SparseTensor.
Pengembalian:
-
Output
: 1-D. NilaiNNZ
untuk hasilSparseTensor
.
Konstruktor dan Destruktor | |
---|---|
SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape) |
Atribut publik | |
---|---|
operation | |
output |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atribut publik
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
SparseSoftmax
SparseSoftmax( const ::tensorflow::Scope & scope, ::tensorflow::Input sp_indices, ::tensorflow::Input sp_values, ::tensorflow::Input sp_shape )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const