當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python tf.linalg.trace用法及代碼示例

計算張量 x 的軌跡。

用法

tf.linalg.trace(
    x, name=None
)

參數

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

返回

  • 輸入張量的軌跡。

trace(x) 返回 x 中每個 inner-most 矩陣的主對角線的總和。如果 x 的秩為 k 且形狀為 [I, J, K, ..., L, M, N] ,則輸出是秩為 k-2 且尺寸為 [I, J, K, ..., L] 的張量,其中

output[i, j, k, ..., l] = trace(x[i, j, k, ..., l,:,:])

例如:

x = tf.constant([[1, 2], [3, 4]])
tf.linalg.trace(x)  # 5

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

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.linalg.trace。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。