當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.linalg.LinearOperatorLowerTriangular.diag_part用法及代碼示例


用法

diag_part(
    name='diag_part'
)

參數

  • name Op 的名稱。

返回

  • diag_part 與自己相同的dtypeTensor

有效地獲取此運算符的 [batch] 對角線部分。

如果此運算符具有形狀 [B1,...,Bb, M, N] ,則返回形狀為 [B1,...,Bb, min(M, N)]Tensor diagonal ,其中 diagonal[b1,...,bb, i] = self.to_dense()[b1,...,bb, i, i]

my_operator = LinearOperatorDiag([1., 2.])

# Efficiently get the diagonal
my_operator.diag_part()
==> [1., 2.]

# Equivalent, but inefficient method
tf.linalg.diag_part(my_operator.to_dense())
==> [1., 2.]

相關用法


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