如何在Numpy / Theano中表达c [i，j，k] = a [i，j] * b [i，k]？

Question

定义

c[i,j,k] = a[i,j] * b[i,k] for any i, j, k

是关于i的元素方面的产品，是关于j和k的外部产品。有没有办法在没有循环的NumPy / Theano中表达这个？

Answer 1

我找到了一个适用于Numpy和Theano的解决方案：

c = a[:, :, np.newaxis] * b[:, np.newaxis, :]