소스 뷰어
조회수 :   1
# 행렬 곱

import numpy as np

# 두 행렬
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 아인슈타인 표기법을 사용한 행렬 곱 계산
matrix_product = np.einsum('ik,kj->ij', a, b)
# 'ik,kj->ij'는 행렬 곱의 의미로, 
# i행의 k열 요소를 곱하여 j에 결과를
# 합산한다는 의미입니다.

print( "a = ", a )
print( "b = ", b )
print("Matrix product:", matrix_product, sep="\n") 
a =  [[1 2]
 [3 4]]
b =  [[5 6]
 [7 8]]
Matrix product:
[[19 22]
 [43 50]]