How to concate two feature(B*C*H*W) tensor together

[Zichun_Zhang (Cipher)]

Hello, How to concatenate two feature(1 * C * H * W) tensor together, without using extra space of GPU,
for instance, just append the second tensor B after the first tensor A, so become the output tensor(1 * 2C * H * W )
Thx!

[albanD]

Hi,

You can use torch.cat([t1, t2], dim=1).

[Zichun_Zhang (Cipher)]

thx a lot