flux tensoriel : : opérations : : SparseSoftmax

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.

Retour:

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

Attributs publics

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