Both S0 and S1 tensors size is [12, 16, 64]
from following for loop, I want to give a result in [12, 16*16, 64] size, however, the result is [12, 16, 17, 64]. What kinds of modification I should apply?
S2 =[]
for i in range(0, 16):
S2.append(torch.cat((S0[:,i,None], S1), dim=1))
S3 = torch.stack(S2,1)
Thanks!