Numpy array 행렬곱 하는 법(np.dot(), @)

안녕하세요 jay입니다.
오늘은 numpy array 사이 행렬곱 하는 법에 대해 알아보도록 하겠습니다.

행렬곱은 @ 연산자 혹은 np.dot()함수로 일어납니다.
행렬곱은 A(행,열) X B(행,열) 을 한다고 가정했을 때
A의 열과 B의 행의 값이 반드시 일치해야 실행됩니다.
A(3,3) X B(3,2)이라고 가정하면 3X2 형태의 행렬이 나타납니다.
아래의 예시를 보시길 바랍니다


# 행렬곱 

import numpy as np

# 행렬끼리 곱을 하려면 @ 혹은 dot함수를 사용하면 된다

arr_1 = np.ones((3,3))
print(arr_1)

arr = np.ones((3,2))*3
print(arr)

print(arr_1@arr)
"""
[[9. 9.]
 [9. 9.]
 [9. 9.]]
"""

print(np.dot(arr_1,arr))

"""
[[9. 9.]
 [9. 9.]
 [9. 9.]]
"""

댓글