anantguptadbl:

ch.Size([3, 10])


if my tensor gave me like that
_embedding shape is  torch.Size([7, 768])
_embedding shape is  torch.Size([8, 768]) … and so on
and i need to make it to sum all different number like 7 , 8 , .... so i need to use this line
torch.cat(_embedding, axis=0) ,right ? as i got the summation only without 768