Hi,
A and B are feature maps, in shape of [12, 9, 64] each.
I have this code to concatenate them 9 times in different cells:
A = A.repeat(1, 1, 9).view(12, 81, 64)
B = B.repeat(1, 9, 1)
C = torch.cat((A, B), dim=2)
However, I faced with lack of memory to run this code. Is it possible to use expand instead of view, here also?
Thanks!