Python tf.raw_ops.DiagPart用法及代码示例

返回张量的对角线部分。

用法

tf.raw_ops.DiagPart(
    input, name=None
)

input 一个Tensor。必须是以下类型之一：bfloat16 , half , float32 , float64 , int32 , int64 , complex64 , complex128。秩 k 张量，其中 k 是偶数而不是零。
name 操作的名称(可选)。

此操作返回带有 input 的 diagonal 部分的张量。 diagonal 部分计算如下：

假设 input 有维度 [D1,..., Dk, D1,..., Dk] ，那么输出是一个秩为 k 的张量，维度为 [D1,..., Dk] 其中：

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik].

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.DiagPart。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。