A and B both are in: [506, 16, 256]
And I want to move through each 4*4 partitions. (repeat: 16 times)
Also, for two for loop;
for i in range(1, 17):
for j in range (1, 17):
y = torch.cat((A[:,i,:], B[:, j, :]), dim =[-2, 1])
I faced this error:
cat(): argument ‘dim’ must be int, not list
Thanks!