tensorflow::ops::SparseSoftmax

#include <sparse_ops.h>

Applique la fonction softmax à un SparseTensor ND par lots.

Résumé

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

Cette opération est équivalente à l'application de la fonction tf.nn.softmax() classique à chaque sous-matrice logique interne de forme [B, C] , à la différence que les éléments implicitement nuls ne sont pas pris en compte . 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 implicitement nuls d'origine ; (3) Renormalise les éléments restants.

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

Arguments :

  • portée : un objet Scope
  • sp_indices : matrice 2D NNZ x R contenant les indices des valeurs non vides dans un SparseTensor, dans l’ordre canonique.
  • sp_values : 1-D. Valeurs non vides NNZ correspondant à sp_indices .
  • sp_shape : 1-D. Forme du tenseur creux d'entrée.

Retours :

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

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