Pytorch中torch.stack()的理解

整理一下两位老哥的博客内容，加深一下印象，感谢两位老哥整理
添加链接描述
添加链接描述

有a，b，c三个tensor.

dim=0，将[]中的n个矩阵变成n维，按照顺序，第i维是第i个矩阵，size=(i，x，y)

dim=1，将列表[]中的每个矩阵的第一行组成第一维矩阵，依次下去,每个数组第n行组成第n维数组。size=(n，i，y)

dim=2，将列表[]中的每个矩阵的第一行的第一个值，组成第一维的第一行，每一个矩阵的第一行第二个值组成第一维的第二行，每个矩阵的第二行的第一个值，组成第二维的第一行，每一个矩阵的第二行第二个值组成第二维的第二行，以此类推

添加链接描述
还有这位老哥的叙述也很详细，结合起来看对**torch.stack()**会有较好的理解