tf.matrix_set_diag()

tf.matrix_set_diag(input,diagonal,name=None)
功能：将输入矩阵的对角元素置换为对角元素。
输入：input：矩阵，diagonal：对角元素。
例：

a=tf.constant([[0,1,2,3],[-1,0,1,2],[-2,-1,0,1],[-3,-2,-1,0]])
z=tf.matrix_set_diag(a,[10,11,12,13])

z==>[[10  1  2  3]
     [-1 11  1  2]
     [0  -1 12  1]
     [0   0 -1 13]]

作者：华夏意匠
链接：https://www.jianshu.com/p/336772764fef
來源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。