tf.linalg.trace

Вычислить след тензора x.

tf.linalg.trace(
    x, name=None
)

trace(x) возвращает сумму по главной диагонали каждого внутреннего матрицы в x. Если x имеет ранг k с формой [I, J, K, ..., L, M, N], то выход является тензором ранга k-2 с измерениями [I, J, K, ..., L] , где

output[i, j, k, ..., l] = trace(x[i, j, i, ..., l, :, :])

Например:

x = tf.constant([[1, 2], [3, 4]])
tf.linalg.trace(x)  # 5

x = tf.constant([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])
tf.linalg.trace(x)  # 15

x = tf.constant([[[1, 2, 3],
                  [4, 5, 6],
                  [7, 8, 9]],
                 [[-1, -2, -3],
                  [-4, -5, -6],
                  [-7, -8, -9]]])
tf.linalg.trace(x)  # [15, -15]