TensorFlow函數(shù)：tf.trace

tf.trace函數(shù)

別名：

• tf.linalg.trace
• tf.trace

tf.trace(
    x,
    name=None
)

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

請參閱指南：數(shù)學(xué)函數(shù)>矩陣數(shù)學(xué)函數(shù)

計(jì)算張量 x 的軌跡(trace).

trace(x) 返回沿 x 中每個(gè)最內(nèi)層矩陣的主對角線的總和.如果 x 是秩的 k 具有形狀[I, J, K, ..., L, M, N],然后將其輸出是一個(gè)張量,它的秩為 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]

函數(shù)參數(shù)：

• x：張量.
• name：操作的名稱(可選).

函數(shù)返回值：

tf.trace函數(shù)返回輸入張量的軌跡(trace).