RuntimeError: The size of tensor a (3) must match the size of tensor b (1024) at non-singleton dimension 2 해결법

RuntimeError: The size of tensor a (3) must match the size of tensor b (1024) at non-singleton dimension 2 해결법

해결법 1

저의 경우 Pytorch에서 torchvision.utils.make_grid를 하는 과정에서 위 문제를 겪었습니다. 보통 batch x channel x height x width의 형식의 4 dim으로 trainloader를 불러오는데 위 문제의 경우 dimension 일치 문제를 해결해야 하기에 batch가 포함된 4차원을 3차원으로 수정 변경함으로써 문제 해결하였습니다.

간단히 말해서 torch.tensor의 squeeze를 통해 해결해봅시다.

해결법 2

github Issue에서 알아본 사실로 해당 dimension에 맞는 위치가 필요할 수도 있다고 전해줍니다.

이에 만약 tensor가 600,600,3 size를 가진다면 3,600,600이 되도록 torch.tensor.view(3,600,600)의 방법을 사용해봄은 어떤가요

https://github.com/pytorch/pytorch/issues/9446