Несколько матричных продуктов в numpy - PullRequest
0 голосов
/ 06 марта 2020

У меня есть два numpy массива A, B фигур (k, m, n) и (k, n, l) соответственно. Я заинтересован в получении следующего numpy массива:

[A[0]@B[0], A[1]@B[1], ..., A[k-1]@B[k-1]]

Очевидно, я могу сделать это с помощью для l oop, но я бы хотелось бы узнать, есть ли более "numpy" способ сделать это, который может работать лучше.

Спасибо.

...