本文主要是介绍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
來源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这篇关于tf.matrix_set_diag()的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!