TensorFlow函数：tf.trace

tf.trace函数

别名：

• tf.linalg.trace
• tf.trace

tf.trace(
    x,
    name=None
)

定义在：tensorflow/python/ops/math_ops.py.

请参阅指南：数学函数>矩阵数学函数

计算张量 x 的轨迹(trace).

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.trace(x)  # 5

x = tf.constant([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])
tf.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.trace(x)  # [15, -15]

函数参数：

• x：张量.
• name：操作的名称(可选).

函数返回值：

tf.trace函数返回输入张量的轨迹(trace).