使用2d张量构造3d对角张量-Java 学习之路

给定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]]] . 我想用它作为我的高斯协方差分数 .

1 回答