在tensorflow中尝试一下：
import tensorflow as tfA = [[1,2],[3 ,4],[5,6]]B = tf.matrix_diag(A)print(B.eval(session=tf.Session()))[[[1 0] [0 2]] [[3 0] [0 4]] [[5 0] [0 6]]]
在tensorflow中尝试一下：
import tensorflow as tfA = [[1,2],[3 ,4],[5,6]]B = tf.matrix_diag(A)print(B.eval(session=tf.Session()))[[[1 0] [0 2]] [[3 0] [0 4]] [[5 0] [0 6]]]
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
