ErrorThe size of tensor a (96) must match the size of tensor b (80) at non-singleton dimension 2
ยท Sign up or log in to comment