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