阅读(7.8k) 书签 (0)

TensorFlow函数:tf.trace

2019-01-31 10:27 更新

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