Error in tensor size mismatch

[SURAJ28092001]

Hi @jacobgil,
I am using this project for my swin transformers but it is giving error showing

     35             # print("a : ",a)
     36             # print(a.size())
---> 37             a = a / a.sum(dim=-1)
     38 
     39             result = torch.matmul(a, result)

RuntimeError: The size of tensor a (49) must match the size of tensor b (64) at non-singleton dimension 1 ```

Please a give a solution for the same

[raphaelattias]

@SURAJ28092001 Same, I am getting this issue with ViT-8

EDIT: setting keep_dim=True fixes the issue.

a = a / a.sum(dim=-1,keepdim=True)