tf.raw_ops.DiagPart

Возвращает диагональную часть тензора.

tf.raw_ops.DiagPart(
    input, name=None
)

Данная операция возвращает тензор с диагональной частью diagonal input. Диагональная часть вычисляется следующим образом:

Предположим, что 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]