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).