torch.stack()

torch.stack() 含义:在指定的维度对元素进行叠加 例子: 使用torch生成如下的三个tensor: a, b, c 在指定的维度 0 对a, b, c进行叠加: 在指定的维度 1 对a, b, c进行叠加: 在指定的维度 2 对a, b, c进行叠加: 注意: 对于维度 0, 1, 2应该从外往里面看进去. 在指定的维度对tensor进行叠加时, 应该把该维度对应的个体看成是一个块,

>>阅读原文<<