TensorFlow如何返回对角线

tf.diag_part

diag_part ( 
    input , 
    name = None
 )

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

返回张量的对角线部分.
此操作返回输入的对角线部分的张量.对角部分计算方法如下:
假设输入(input)具有维度:[D1,..., dk, D1,..., dk],那么输出是一个秩为 k 的张量,并且其维度为:[D1,..., dk] :

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik]

例如:

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

ARGS:

  • input:一个张量.必须是下列类型之一:float32,float64,int32,int64,complex64,complex128.秩为 k 的张量的取值是 2, 4, 或6.
  • name:操作的名称(可选).

返回:

返回张量.与 input 具有相同的类型,为提取的对角线.

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

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部