2020-08-20 10:40:49

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

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import torch
 
#%% 행렬 곱셈
= torch.FloatTensor([[12],
                       [34]])
 
= torch.FloatTensor([[1], 
                       [2]])
 
print("행렬 A의 크기: ", A.shape)
print("행렬 B의 크기: ", B.shape)
 
print("A x B: \n", A.matmul(B))
 
#%% element-wise 곱셈
= torch.FloatTensor([[12],
                       [34]])
 
= torch.FloatTensor([[11],
                       [00]])
 
print("C와 D의 element-wise 곱셈: \n", C.mul(D))
cs

 

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