TensorFlow函数:tf.matrix_set_diag

由 Carrie 创建, 最后一次修改 2017-11-01
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.

以上内容是否对您有帮助:
二维码
建议反馈
二维码