【发布时间】：2017-08-10 22:25:43
【问题描述】：
给定 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]]] .我想用它作为我的高斯协方差矩阵。
【问题讨论】：
标签： python numpy matrix tensorflow broadcast
【发布时间】：2017-08-10 22:25:43
【问题描述】：
好的，我想通了。 tf.matrix_diag() 可以解决问题...
【讨论】：