У меня есть массив A x B и еще один массив D x A x A, и я пытаюсь найти эффективные способы вычисления суммы скалярных произведений двух массивов вдоль оси D (так, чтобы результат был A x B множество). Самый очевидный способ - использовать цикл for:
result = np.zeros(first_array.shape)
for d in range(0,second_array.shape[0]):
result = result + np.dot(second_array[d], first_array)
print result
Мне интересно, есть ли более эффективные способы вычисления этого в numpy. Я немного прочитал _10 _ но, к сожалению, не совсем понимаю, сможет ли он помочь в этом случае.
