RuntimeError: Invalid index in gather at /pytorch/aten/src/TH/generic/THTensorEvenMoreMath.cpp:657

[Meimeiainaonao]

Thanks for your sharing. I don't know why I change the input size like 56x56, It is can meet this question.

I print the index.shape is torch.Size([8, 32, 28224]).

Please help me!!