Numpy Dot Произведение двух двухмерных массивов в numpy для получения трехмерного массива

Question

Извините за плохо объясненный заголовок. Я пытаюсь распараллелить часть своего кода и застрял на точечном продукте. Я ищу эффективный способ сделать то, что делает код ниже, я уверен, что есть простое решение линейной алгебры, но я очень застрял:

puy = np.arange(8).reshape(2,4)
puy2 = np.arange(12).reshape(3,4)

print puy, '\n'
print puy2.T

zz = np.zeros([4,2,3])

for i in range(4):
    zz[i,:,:] = np.dot(np.array([puy[:,i]]).T,
                np.array([puy2.T[i,:]]))

Answer 1

Одним из способов было бы использовать np.einsum, что позволяет указать, что вы хотите, чтобы произошло с индексами:

>>> np.einsum('ik,jk->kij', puy, puy2)
array([[[ 0,  0,  0],
        [ 0, 16, 32]],

       [[ 1,  5,  9],
        [ 5, 25, 45]],

       [[ 4, 12, 20],
        [12, 36, 60]],

       [[ 9, 21, 33],
        [21, 49, 77]]])
>>> np.allclose(np.einsum('ik,jk->kij', puy, puy2), zz)
True

Answer 2

Вот еще один способ с broadcasting -

(puy[None,...]*puy2[:,None,:]).T