Torch.cat()基础

先贴一段官方文档：

 

torch.cat主要作用是将两个不同的tensor拼接起来，其中要注意的是第三个参数“dim”。

dim = 0 可以理解为“横着拼”，即按照维度0拼接；

同理dim = 1 可以理解为“竖着拼”，即按照维度1拼接；