[PyTorch] 파이토치에서 행렬 곱셈과 element-wise 곱셈

파이토치에서 행렬 곱셈은 matmul 메소드로, 행렬 원소별(element-wise) 곱셈은 mul 메소드로 가능합니다. 사용법은 다음 예제를 통해 확인하세요.

 

import torch
 
#%% 행렬 곱셈
A = torch.FloatTensor([[1, 2],
                       [3, 4]])
 
B = torch.FloatTensor([[1], 
                       [2]])
 
print("행렬 A의 크기: ", A.shape)
print("행렬 B의 크기: ", B.shape)
 
print("A x B: \n", A.matmul(B))
 
#%% element-wise 곱셈
C = torch.FloatTensor([[1, 2],
                       [3, 4]])
 
D = torch.FloatTensor([[1, 1],
                       [0, 0]])
 
print("C와 D의 element-wise 곱셈: \n", C.mul(D))

 

위 코드를 실행하면 다음과 같은 화면이 출력됩니다.