Python tf.linalg.tensor_diag用法及代碼示例

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

用法

tf.linalg.tensor_diag(
    diagonal, name=None
)

參數

diagonal 一個Tensor。必須是以下類型之一：bfloat16 , half , float32 , float64 , int32 , int64 , complex64 , complex128。秩 k 張量，其中 k 最多為 1。
name 操作的名稱(可選)。

一個Tensor。具有與 diagonal 相同的類型。

給定 diagonal ，此操作返回一個帶有 diagonal 的張量，其他所有內容都用零填充。對角線計算如下：

假設 diagonal 的維度為 [D1,..., Dk]，那麽輸出是一個秩為 2k 的張量，維度為 [D1,..., Dk, D1,..., Dk]，其中：

output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik]和 0 其他地方。

例如：

# '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]]

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.linalg.tensor_diag。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。