torch.stack解析以及与torch.cat的关系

torch.stack这个函数看上去输出的结果很奇怪，但是其实解析之后很简单。 举个例子： 然后使用torch.stack函数： 一眼看上去，好像有点懵逼，不急，我们慢慢解析。 这个函数是这样的，他的dim参数表明他会在dim新建一个维度，然后在这个维度上进行concat。 因此，我们可以这么理解，假如取dim=1，那么我们可以将a，b同时在dim=1新增一个维度，即： 然后再在dim=1的维度上

>>阅读原文<<