Jingles:

x = torch.randn((7, 32)) # 1. shape's batch is 7, because this is last batch, and I dont want to "drop_last"



Intentionally want to avoid drop_last=True.
My current solution is:
x = F.pad(x, (0,0,0,(self.batchsize-b)))