使用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（）可以解決問題...