torch.matmul() 에 대해서 이해해보자.

motivation: torch.matmul() 에 대해서 이해해보자.

1. 입력 tensor가 3차원인 경우 –> torch.matmul(A,B) 는, A가 (b, n, m) 차원을 가지고, B가 (b, m, p)의 차원을 가지면, 결과는 (b,n,p) 차원을 가지는 tensor가 나오게 된다.

2. 입력 tensor가 4차원 이상인 경우 –> torch.matmul(A,B) 는, A가 (a, b, c, n, m) 차원을 가지고, B가 (a, b, c, m, p)의 차원을 가지면, 결과는 (a,b,c,n,p) 차원을 가지는 tensor가 나오게 된다.

즉 마지막 두개의 차원만 제외하고 모두 차원이 유지된다고 할 수 있다.