python – 2d矩阵的Theano矩阵乘法给出3d矩阵

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

我已经编写了以下代码(根据numpy案例的建议),假设广播也在theano中工作:

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

但它抛出以下错误:

AsTensorError: ('Cannot convert Ellipsis to TensorType', <type 'ellipsis'>)
最佳答案
根据@Divakar的评论,将其更改为

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

只需将其作为答案发布,以便人们知道答案就是这样.

转载注明原文:python – 2d矩阵的Theano矩阵乘法给出3d矩阵 - 代码日志