TensorDiag

מחזירה טנזור אלכסוני עם ערכי אלכסון נתונים.

בהינתן `אלכסון`, פעולה זו מחזירה טנזור עם `אלכסון` וכל השאר מרופד באפסים. האלכסון מחושב באופן הבא:

נניח של'אלכסון' יש מידות [D1,..., Dk], אז הפלט הוא טנסור בדרגה 2k עם מידות [D1,..., Dk, D1,..., Dk] כאשר:

`פלט[i1,..., ik, i1,..., ik] = אלכסון[i1, ..., ik]` ו-0 בכל מקום אחר.

לְדוּגמָה:

# 'diagonal' is [1, 2, 3, 4]
 tf.diag(diagonal) ==> [[1, 0, 0, 0]
                        [0, 2, 0, 0]
                        [0, 0, 3, 0]
                        [0, 0, 0, 4]]