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 dengan sp_indices .
  • sp_bentuk: 1-D. Bentuk masukan SparseTensor.

Pengembalian:

  • Output : 1-D. Nilai NNZ untuk hasil SparseTensor .

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