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

2023年3月20日 208次阅读

我想完成以下问题在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,:]

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

点赞