当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.linalg.LinearOperatorKronecker.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.LinearOperatorKronecker.diag_part。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。