torch.repeat() lets us specify the number of times to repeat for each dimension.
Something like below should work:
>>> import torch
>>> x = torch.randn(2,2)
>>> y = x.repeat(1, 32)
>>> y.shape
torch.Size([2, 64])
1 Like