Pytorch 中 tensor的维度拼接

torch.stack() 和 torch.cat() 都可以按照指定的维度进行拼接，但是两者也有区别，torch.satck() 是增加新的维度进行堆叠，即其维度拼接后会增加一个维度；而torch.cat() 是在原维度上进行堆叠，即其维度拼接后的维度个数和原来一致。具体说明如下： torch.s