У меня есть Матрица A с формой (2,2, N) и Матрица V с формой (2, N)
Я хочу векторизовать следующее:
F = np.zeros(N)
for k in xrange(N):
F[k] = np.dot( A[:,:,k], V[:,k] ).sum()
В любом случае это можно сделать с помощью tenordot или любой другой функции numpy без явного зацикливания?