带广播的张量产品

时间:2016-06-22 15:29:58

标签: matrix theano

假设我们有一个大小为A的矩阵m x n和一个大小为B的张量k x s x n x t。我想将这个张量和矩阵相乘。当我运行命令theano.tensor.dot(A,B)时,输出的形状将是m x k x s x t,这意味着Theano以错误的方式广播2D矩阵。

我想输出大小k x s x m x t意味着只有张量的最后两个维度乘以矩阵。我该如何解决这个广播问题?

0 个答案:

没有答案