본문 바로가기

Language/Python

Python 행렬 연산 np.dot()

728x90

 

 

딥러닝 신경망에서 행렬의 곱으로 신경망 연산을 할 때 np.dot()을 쓸때가 있습니다.

 

[1, 2]와 [[1, 3, 5], [2, 4, 6]] 을 곱할 때 np.dot()으로 연산을 할 수 있습니다.

 

그러면 다음과 같은 연산이 이루어 지고 (1 * 1) + (2 * 2), (1 * 3) + (2 * 4), (1 * 5) + (2 * 6)

 

결과는 [5,  11, 17]이 됩니다.

 

위 연산을 코드로 표현 하면

import numpy as np

x = np.array([1, 2])
w = np.array([[1, 3, 5], [2, 4, 6]])

y = np.dot(x, w)
print(y)

결과

[ 5 11 17 ]

 

2x2 * 2x2는 아래와 같습니다.

import numpy as np

x = np.array([[1, 2],[3, 4]])
y = np.array([[5, 6],[7, 8]])

result = np.dot(x, y)
print(result)

 

결과

[[19 22]

 [43 50]]

 

end.

728x90
블로그 주인장입니다. 원하시는 정보는 얻으셨나요? 이 포스트에서 추가로 필요한 정보가 있으시면 여기에 남겨주세요.