Concat two tensors
I want to concat two tensors of size a: torch.Size([16, 1]) and b: torch.Size([16, 120])
to be of size torch.Size([16, 121])
could you please help with that?

Here you can use the torch.cat() function.
Example:
&gt;&gt;&gt; a = torch.rand([16,1])
&gt;&gt;&gt; b = torch.rand([16,120])
&gt;&gt;&gt; a.size()
torch.Size([16, 1])
&gt;&gt;&gt; b.size()
torch.Size([16, 120])
&gt;&gt;&gt; c = torch.cat((a,b),dim=1)
&gt;&gt;&gt; c.size()
torch.Size([16, 121])

What you want to do is to concatenate the tensors on the first dimension (dim=1).
