יום הקהילה ML הוא 9 בנובמבר! הצטרפו אלינו עדכונים מ- TensorFlow, JAX, ועוד למידע נוסף

זרימת טנסור :: אופ :: DiagPart

#include <array_ops.h>

מחזיר את החלק האלכסוני של הטנזור.

סיכום

פעולה זו מחזירה טנזור עם החלק diagonal של input . החלק diagonal מחושב באופן הבא:

נניח input יש ממדים [D1,..., Dk, D1,..., Dk] , אז הפלט הוא מותח של דרגה k עם ממדים [D1,..., Dk] שם:

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik] .

לדוגמה:

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

טענות:

  • היקף: אובייקט Scope
  • קלט: דרג k טנסור כאשר k הוא שווה ולא אפס.

החזרות:

  • Output : האלכסון שחולץ.

קונסטרוקטורים ומשחתנים

DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input)

תכונות ציבוריות

diagonal
operation

פונקציות ציבוריות

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

תכונות ציבוריות

אֲלַכסוֹנִי

::tensorflow::Output diagonal

מבצע

Operation operation

פונקציות ציבוריות

DiagPart

 DiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

צוֹמֶת

::tensorflow::Node * node() const 

אופרטור :: זרימת טנסור :: קלט

 operator::tensorflow::Input() const 

אופרטור :: זרימת טנסור :: פלט

 operator::tensorflow::Output() const