What is Pytorch(행렬곱)?

PyTorch에서 텐서의 행렬 곱셈(또는 텐서의 곱셈) 연산은 다양한 형태로 수행할 수 있으며, 이 연산은 선형 대수학의 기본적인 연산 중 하나이다.

 

행렬 곱셈의 기본 개념

행렬 곱셈은 두 행렬을 결합하여 새로운 행렬을 생성하는 연산이다. 행렬 A와 B가 있을 때, A의 열 수가 B의 행 수와 같아야 곱셈이 가능하다.

• 행렬 A: 크기 M x N
• 행렬 B: 크기 N x P
• 곱셈 결과: 크기 M x P

행렬 C의 각 원소는 다음과 같이 계산된다:

A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([[5.0, 6.0], [7.0, 8.0]])

#matrix multiplication
C1 = torch.mm(A, B) 
C2 = A @ B
C3 = torch.matmul(A, B)  

print(C1)
print(C2)
print(C3)