I about me

[선형대수학] 코드와 함께하는 행렬 연산 본문

AI/선형대수학

[선형대수학] 코드와 함께하는 행렬 연산

ssungni 2024. 10. 8. 20:42
A = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

B = np.array([
    [0, 1, 2],
    [2, 0, 1],
    [1, 2, 0]
])

덧셈

A + B
# arrray([1, 3, 5]
#       [6, 5, 7]
#       [8, 10, 9])

스칼라곱

5 * A
# arrray([5, 10, 15]
#       [20, 25, 30]
#       [35, 40, 45])

두 행렬의 곱

A @ B

요소별 곱하기

A * B
# arrray([0, 2, 6]
#       [8, 0, 6]
#       [7, 16, 0])