tensorflow::ops::SparseSoftmax

#include <sparse_ops.h>

Menerapkan softmax pada ND SparseTensor yang dikelompokkan.

Ringkasan

Input tersebut 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() biasa ke setiap submatriks logis terdalam dengan bentuk [B, C] , tetapi dengan catatan bahwa elemen yang secara implisit bernilai nol tidak ikut serta . Secara spesifik, algoritma ini setara dengan berikut ini:

(1) Menerapkan tf.nn.softmax() ke tampilan yang diperpadat dari setiap submatriks terdalam dengan bentuk [B, C] , sepanjang dimensi ukuran-C; (2) Menutupi lokasi implisit nol asli; (3) Menormalisasi ulang elemen yang tersisa.

Oleh karena itu, hasil SparseTensor memiliki indeks dan bentuk bukan nol yang persis sama.

Argumen:

  • Cakupan: Objek Cakupan
  • sp_indices: Matriks 2-D. NNZ x R dengan indeks nilai non-kosong dalam SparseTensor, dalam urutan kanonik.
  • sp_values: 1-D. NNZ nilai non-kosong yang sesuai dengan sp_indices .
  • sp_shape: 1-D. Bentuk SparseTensor masukan.

Mengembalikan:

  • 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::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const