tensoreflusso:: ops:: SparseSoftmax

#include <sparse_ops.h>

Applica softmax a un ND SparseTensor in batch.

Riepilogo

Gli input rappresentano un ND SparseTensor con forma logica [..., B, C] (dove N >= 2 ), e con indici ordinati nell'ordine lessicografico canonico.

Questa operazione equivale ad applicare la normale tf.nn.softmax() a ciascuna sottomatrice logica più interna con forma [B, C] , ma con il problema che gli elementi implicitamente zero non partecipano . Nello specifico, l'algoritmo è equivalente al seguente:

(1) Applica tf.nn.softmax() a una vista più densa di ciascuna sottomatrice più interna con forma [B, C] , lungo la dimensione size-C; (2) Maschera le posizioni originali implicitamente zero; (3) Rinormalizza gli elementi rimanenti.

Quindi, il risultato SparseTensor ha esattamente gli stessi indici e la stessa forma diversi da zero.

Argomenti:

  • scope: un oggetto Scope
  • sp_indices: 2-D. Matrice NNZ x R con gli indici dei valori non vuoti in uno SparseTensor, in ordinamento canonico.
  • valori_sp: 1-D. NNZ valori non vuoti corrispondenti a sp_indices .
  • sp_forma: 1-D. Forma dello SparseTensor in ingresso.

Ritorna:

  • Output : 1-D. I valori NNZ per il risultato SparseTensor .

Costruttori e distruttori

SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape)

Attributi pubblici

operation
output

Funzioni pubbliche

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Attributi pubblici

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

SparseSoftmax

 SparseSoftmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input sp_indices,
  ::tensorflow::Input sp_values,
  ::tensorflow::Input sp_shape
)

nodo

::tensorflow::Node * node() const 

operatore::tensorflow::Input

 operator::tensorflow::Input() const 

operatore::tensorflow::Output

 operator::tensorflow::Output() const