본문 바로가기

Python/AI 수학 with Python

[Python] 벡터의 내적 표시

numpy의 dot() 함수로 간단하게 구현을 할 수 있다.

또한, 각각의 요소를 곱한 후 더하는 방식으로도 구현을 할 수 있다.

[In]

import numpy as np

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

print('------ dot() 함수 ------')
print(np.dot(a, b))
print('------ sum() 함수 ------')
print(np.sum(a * b)) # 각각의 요소끼리 곱한 후 더한다.

[Out]

------ dot() 함수 ------
10
------ sum() 함수 ------
10