TensorDiagPart

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

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

נניח של-input' יש ממדים `[D1,..., Dk, D1,..., Dk]`, אז הפלט הוא טנסור בדרגה `k` עם מידות `[D1,..., Dk]` אֵיפֹה:

`אלכסון[i1,..., ik] = קלט[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]