tf.matrix_set_diag 函数

matrix_set_diag(
    input,
    diagonal,
    name=None
)

参考指南：数学函数>矩阵数学函数

返回具有新的批处理对角值的批处理的矩阵张量.

在该 TensorFlow 函数中分别给定一个 input 和 diagonal,此操作返回与 input 具有相同的形状和值的张量,最内层矩阵的主对角线除外.它们将被对角线上的值覆盖.

输出的计算如下：

假设 input 具有 k+1 维度 [I, J, K, ..., M, N],并且 diagonal 具有 k 维度 [I, J, K, ..., min(M, N)].那么就会输出一个秩为 k+1 的张量,其维度为 [I, J, K, ..., M, N]：

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] for m == n
  
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] for m != n

参数：

• input：张量；秩为 k+1,并且 k >= 1.
• diagonal：张量.必须与 input 具有相同的类型；秩为 k,并且 k >= 1.
• name：操作的名称(可选).

返回值：

该函数将返回一个张量,与 input 具有相同的类型,秩为 k+1,并且 output.shape = input.shape.