使用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（）可以解决问题...