tf.diag_part学习

diag_part(
    input,
    name=None
)

返回张量的对角线部分。

该操作返回与该diagonal部分的张量input。该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]

input：A Tensor。必须是下列类型之一：float32，float64，int32，int64，complex64，complex128。Rank k张量，其中k是2,4或6。
name：操作的名称（可选）。

import tensorflow as tf


diagonal=[[1,0,0,0],
 [0,2,0,0],
 [0,0,3,0],
 [0,0,0,4]]
with tf.Session() as sess:
     print(sess.run(tf.diag_part(diagonal)))

实验结果：

[1 2 3 4]

程序员有偿接单和程序指导加QQ:734564390