用法
diag_part(
name='diag_part'
)
參數
-
name
此Op
的名稱。
返回
-
diag_part
與自己相同的dtype
的Tensor
。
有效地獲取此運算符的 [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.]
相關用法
- Python tf.linalg.LinearOperatorPermutation.solve用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation.assert_non_singular用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation.matmul用法及代碼示例
- Python tf.linalg.LinearOperatorPermutation用法及代碼示例
- Python tf.linalg.LinearOperatorFullMatrix.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz.solve用法及代碼示例
- Python tf.linalg.LinearOperatorIdentity.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorKronecker.diag_part用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorBlockLowerTriangular.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorLowerTriangular.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant2D.solve用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant3D.diag_part用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz.solvevec用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant2D.assert_non_singular用法及代碼示例
- Python tf.linalg.LinearOperatorToeplitz用法及代碼示例
- Python tf.linalg.LinearOperatorCirculant2D.matvec用法及代碼示例
- Python tf.linalg.LinearOperatorTridiag.solvevec用法及代碼示例
注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.linalg.LinearOperatorPermutation.diag_part。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。