torch.stack()解析

torch.stack()是将原来的几个tensor按照一定方式进行堆叠，然后在按照堆叠后的维度进行切分。

有a，b，c三个tensor.

dim=0

dim=1

dim=2

参考：torch.stack(), torch.cat()用法详解