I have a Variable x
with shape 1x2048x2x2
and another y
shape of 1x2048
, and I need to do a torch.cat
operation between them. The current solution is
y = y.unsqueeze(-1)
y = y.unsqueeze(-1)
y = y.expand_as(x)
I wonder if there is a more elegant way to solve the problem? Thanks