使用diag返回對角張量

tf.diag

diag(
    diagonal,
    name=None
)

參考指南：數(shù)學(xué)>數(shù)學(xué)矩陣函數(shù)

返回具有給定對角線值的對角張量.

給定一個diagonal(對角線),這個操作返回一個帶對角線的張量,其他的都用零填充.

對角線的計算方法如下：

假設(shè) diagonal(對角線)具有維度：[D1,...,Dk],則輸出是具有維度：[D1,...,Dk,D1,...,Dk]的秩為 2k 的張量,其中：

output [i1,..., ik, i1,..., ik] = diagonal [i1, ..., ik] 和其他的都用零填充.

例如：

# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
                       [0, 2, 0, 0]
                       [0, 0, 3, 0]
                       [0, 0, 0, 4]]

ARGS：

• diagonal：張量.必須是下列類型之一：float32,float64,int32,int64,complex64,complex128.秩為 k 的張量,k 最多為 3.
• name：操作的名稱(可選).

返回：

返回一個張量,具有與 diagonal(對角線)相同的類型.