tensoreflusso:: ops:: SparseSoftmax

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.

Resi:

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

Attributi pubblici

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const