The Runtime error:
RuntimeError: The size of tensor a (5) must match the size of tensor b (32) at non-singleton dimension 1
please check the dimensions first. May you please tell me in which line you are getting this error.