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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。