用法
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.LinearOperatorKronecker.matvec用法及代码示例
- Python tf.linalg.LinearOperatorKronecker.matmul用法及代码示例
- Python tf.linalg.LinearOperatorKronecker.solve用法及代码示例
- Python tf.linalg.LinearOperatorKronecker.solvevec用法及代码示例
- Python tf.linalg.LinearOperatorKronecker.assert_non_singular用法及代码示例
- Python tf.linalg.LinearOperatorKronecker用法及代码示例
- Python tf.linalg.LinearOperatorFullMatrix.matvec用法及代码示例
- Python tf.linalg.LinearOperatorToeplitz.solve用法及代码示例
- Python tf.linalg.LinearOperatorIdentity.solvevec用法及代码示例
- Python tf.linalg.LinearOperatorPermutation.solve用法及代码示例
- 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.LinearOperatorPermutation.diag_part用法及代码示例
- Python tf.linalg.LinearOperatorToeplitz用法及代码示例
- Python tf.linalg.LinearOperatorCirculant2D.matvec用法及代码示例
注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.linalg.LinearOperatorKronecker.diag_part。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。