标签: 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矩阵。
A
m x n
B
k x s x n x t
theano.tensor.dot(A,B)
m x k x s x t
我想输出大小k x s x m x t意味着只有张量的最后两个维度乘以矩阵。我该如何解决这个广播问题?
k x s x m x t