소스 뷰어
# 행렬 곱
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")