tensorflow:: ops:: SparseSoftmax

#include <sparse_ops.h>

מחיל softmax על ND SparseTensor שעבר אצווה.

תַקצִיר

הקלטים מייצגים ND SparseTensor בעל צורה לוגית [..., B, C] (כאשר N >= 2 ), ועם אינדקסים ממוינים בסדר הלקסיקוגרפי הקנוני.

פעולה זו שקולה ליישום הפונקציה הנורמלית tf.nn.softmax() על כל תת-מטריצת לוגית פנימית ביותר עם צורה [B, C] , אך עם התפיסה שאלמנטים אפסיים באופן מרומז אינם משתתפים . באופן ספציפי, האלגוריתם שקול לפונקציות הבאות:

(1) מפעיל את הפונקציה tf.nn.softmax() על תצוגה צפופה של כל תת-מטריצה ​​פנימית ביותר עם צורה [B, C] , לאורך ממד הגודל-C; (2) מסווה את המיקומים המקוריים בעלי אפס באופן מרומז; (3) מנרמל מחדש את האלמנטים הנותרים.

לכן, לתוצאת SparseTensor יש בדיוק אותם אינדקסים וצורה שאינם אפס.

טיעונים:

  • היקף: אובייקט היקף
  • sp_indices: מטריצה ​​דו-ממדית NNZ x R עם האינדקסים של ערכים שאינם ריקים ב-SparseTensor, בסדר קנוני.
  • sp_values: ערכים חד-ממדיים שאינם ריקים NNZ התואמים ל- sp_indices .
  • sp_shape: צורה חד-ממדית של SparseTensor הקלט.

החזרות:

  • Output : 1-D. ערכי ה- NNZ עבור התוצאה SparseTensor .

בנאים והורסים

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

מאפיינים ציבוריים

operation
output

תפקידים ציבוריים

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

מאפיינים ציבוריים

מִבצָע 

Operation operation

תְפוּקָה 

::tensorflow::Output output

תפקידים ציבוריים

SparseSoftmax 

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

צוֹמֶת 

::tensorflow::Node * node() const

אופרטור::tensorflow::קלט 

 operator::tensorflow::Input() const

אופרטור::tensorflow::פלט 

 operator::tensorflow::Output() const