使用2d张量构造3d对角张量

Question

给定A = [[1,2]，[3,4]，[5,6]]。如何使用tf.diag（）构造一个3d张量，其中每个堆栈是使用A的值的2d对角矩阵？所以输出应该是B = [[[1,0]，[0,2]]，[[3,0]，[0,4]]，[[5,0]，[0,6]]] 。我想用这个作为我的高斯协方差分数。

Answer 1

好的我明白了。 tf.matrix_diag（）可以解决这个问题......