flux tensoriel : : opérations : : SparseSoftmax

#include <sparse_ops.h>

Applique softmax à un ND SparseTensor par lots.

Résumé

Les entrées représentent un ND SparseTensor avec une forme logique [..., B, C] (où N >= 2 ) et avec des indices triés dans l'ordre lexicographique canonique.

Cette opération équivaut à appliquer le tf.nn.softmax() normal à chaque sous-matrice logique la plus interne avec la forme [B, C] , mais avec le problème que les éléments implicitement nuls ne participent pas . Plus précisément, l'algorithme est équivalent à ce qui suit :

(1) Applique tf.nn.softmax() à une vue densifiée de chaque sous-matrice la plus interne de forme [B, C] , le long de la dimension de taille C ; (2) Masque les emplacements d'origine implicitement nuls ; (3) Renormalise les éléments restants.

Par conséquent, le résultat SparseTensor a exactement les mêmes indices et la même forme non nuls.

Arguments :

  • scope : un objet Scope
  • sp_indices : 2-D. Matrice NNZ x R avec les indices de valeurs non vides dans un SparseTensor, dans l'ordre canonique.
  • sp_values : 1-D. Valeurs NNZ non vides correspondant à sp_indices .
  • sp_shape : 1-D. Forme du SparseTensor d’entrée.

Retours :

  • Output : 1-D. Les valeurs NNZ pour le résultat SparseTensor .

Constructeurs et Destructeurs

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

Attributs publics

operation
output

Fonctions publiques

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

Attributs publics

opération 

Operation operation

sortir 

::tensorflow::Output output

Fonctions publiques

SparseSoftmax 

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

nœud 

::tensorflow::Node * node() const

opérateur :: tensorflow :: Entrée 

 operator::tensorflow::Input() const

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const