Как выразить c[i,j,k] = a[i,j] * b[i,k] в Numpy/Theano?

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, :]