从Numpy或Tensorflow中的线性数组矢量化创建对角正方形数组

从Numpy或Tensorflow中的线性数组矢量化创建对角正方形数组

在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]]]