TensorFlow:tf.matrix_diag函数

tf.matrix_diag 函数
matrix_diag(
    diagonal,
    name=None
)

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

返回具有给定批对角线值的批对角张量.

给定一个对角线(diagonal),这个操作将返回一个带有对角线的张量,而其他所有的东西都加0.对角线的计算如下:

假设对角线具有 k 维度 [I, J, K, ..., N],则输出是秩为 k+1 且维度为 [I,J,K,...,N,N] 的张量,其中:

output[i, j, k, ..., m, n] = 1{m=n} * diagonal[i, j, k, ..., n]

例如:

# 'diagonal' is [[1, 2, 3, 4], [5, 6, 7, 8]]

and diagonal.shape = (2, 4)

tf.matrix_diag(diagonal) ==> [[[1, 0, 0, 0]
                                     [0, 2, 0, 0]
                                     [0, 0, 3, 0]
                                     [0, 0, 0, 4]],
                                    [[5, 0, 0, 0]
                                     [0, 6, 0, 0]
                                     [0, 0, 7, 0]
                                     [0, 0, 0, 8]]]

which has shape (2, 4, 4)

参数:

  • diagonal:张量;秩为 k,并且 k >= 1.
  • name:操作的名称(可选).

返回值:

该函数将返回张量,该张量与 diagonal 具有相同的类型;秩为 k+1,并且 output.shape = diagonal.shape + [diagonal.shape[-1]].

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部