(pytorch) size mismatch

RuntimeError: size mismatch, m1: [32 x 7], m2: [6 x 400] at /pytorch/aten/src/TH/generic/THTensorMath.cpp:136
이 에러가 발생했는데,

m1은 이전 단계의 output, m2는 다음 단계의 input이라고 한다.
m1과 m2를 행렬곱 연산을 하는데
m1 = a x b
m2 = c x d 일때

b == c 이어야 한다

 

참고한 페이지 : https://bblib.net/entry/cnn-%ED%8C%8C%EB%9D%BC%EB%AF%B8%ED%84%B0-%EC%88%98-%EA%B3%84%EC%82%B0-%EB%B0%A9%EB%B2%95-%EB%B0%8F-%EA%BF%80%ED%8C%81-size-mismatch-%EC%97%90%EB%9F%AC-%EA%B4%80%EB%A0%A8