이 행렬곱 연산은 신경망 구현에 핵심이 되는 연산이므로 잘 알아두자
행렬곱 연산은 통계학과 때 배운 것처럼 동일하다. 수식적으로는 이해가 되었기 때문에 바로 코드로 넘어가 보자
행렬곱 연산
두 2차원 tensor D,E가 있고 행렬곱을 하고 싶다면
D.matmul(E) or D.mm(E) or D@E
사용 예시(흑백이미지)
G = torch.tensor([[255, 114, 140],
[39, 255, 46],
[61, 29, 255]])
-
흑백 이미지의 좌우로 대칭이동
H = torch.tensor([[0, 0, 1],
[0, 1, 0],
[1, 0, 0]])
H라는 tensor를 만들어서 G@H를 해주면 이미지가 좌우로 대칭이동하게 된다.
-
흑백 이미지의 상하로 대칭이동
이번에 H@G를 해주면 이미지가 상하로 대칭이동하게된다.
