Numpy 두 행렬, 행의 pairwise 내적

Question

현재 우리는 python 프로젝트를 수행 중이며 성능 제약으로 인해 많은 벡터화가 필요합니다. 우리는 다음과 같은 계산으로 끝난다 : 우리는 모양이 두 개의 numpy 어레이 인 (20,6)을 가지고 행의 pairwise 내적을 계산하려고한다. 즉, 각 행은 각각의 벡터에 의해 얻어진 스칼라 인 마지막에 (20,1) 행렬을 얻어야한다. 도트 곱셈.

Answer 1

당신은 현명한 두 개의 배열 요소를 곱하면 다음 행으로 합계를 수행 한 후 각 요소는이 개 원래 배열의 행에서 내적 인 배열을 가질 수 있습니다

a = np.array([[1,2], [3,4]]) 
b = np.array([[3,4], [2,1]]) 

(a * b).sum(axis=1) 
# array([11, 10])