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

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页