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
來源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。