RuntimeError: The expanded size of the tensor (16320) must match the existing size (6375) at non-singleton dimension 1. Target sizes: [4, 16320, 4]. Tensor sizes: [4, 6375, 1]

Experts, I have checked some data, some people say it is the pytorch version of the problem, I am not very clear about this, may I ask what valuable advice?

Hi,

It looks like the Tensor given to the criterion have sizes that do not match.
What is your criterion? What are the size of the Tensors that should be given to it?