python - Theano矩阵乘以2d矩阵得到3d矩阵

Question

我想做的正是以下问题在 numpy 中试图做的事情:
Numpy matrix multiplication of 2d matrix to give 3d matrix

我已经编写了以下代码(根据 numpy 案例的建议)，假设 broadcast 在 theano 中也有效:

y = T.dmatrix('y')
x = T.dmatrix('x')
z = x[...,None]*y[:,None,:]

但它抛出以下错误:

AsTensorError: ('Cannot convert Ellipsis to TensorType', <type 'ellipsis'>)

Answer 1

根据@Divakar 的评论，将其更改为

x[:,:,None] * y[:,None,:]

只是将其作为答案发布，以便人们知道对此有答案。