tensorflow :: opérations :: SparseSoftmax

Applique softmax à un ND SparseTensor par lots.

Résumé

Les entrées représentent un ND SparseTensor de 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 la capture que les éléments implicitement zéro 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 intérieure de forme [B, C] , le long de la dimension 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 forme non nuls.

Arguments:

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

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