TensorFlow函數(shù)：tf.matrix_set_diag

tf.matrix_set_diag 函數(shù)

matrix_set_diag(
    input,
    diagonal,
    name=None
)

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

返回具有新的批處理對角值的批處理的矩陣張量.

在該 TensorFlow 函數(shù)中分別給定一個 input 和 diagonal,此操作返回與 input 具有相同的形狀和值的張量,最內(nèi)層矩陣的主對角線除外.它們將被對角線上的值覆蓋.

輸出的計算如下：

假設(shè) 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

參數(shù)：

• input：張量；秩為 k+1,并且 k >= 1.
• diagonal：張量.必須與 input 具有相同的類型；秩為 k,并且 k >= 1.
• name：操作的名稱(可選).

返回值：

該函數(shù)將返回一個張量,與 input 具有相同的類型,秩為 k+1,并且 output.shape = input.shape.