У меня есть два 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" способ сделать это, который может работать лучше.
Спасибо.