torch.stack()

torch.stack的使用
torch.stack([a,b,c],0)
torch.stack([a,b,c],[1])
torch.stack([a,b,c],[2])
torch.sack([],dim) 设有矩阵为n行y列,有i个矩阵
首先[]list里面的矩阵必须是一样shape的
dim=0时，则就是讲[]中的n个矩阵变成n维，按照顺序，第i维是第i个矩阵，size=(i，x，y)
dim=1是，将列表[]中的每个矩阵的第一行组成第一维矩阵，依次下去,每个数组第n行组成第n维数组。size=(n，i，y)
dim=2是，讲列表[]中的

• 每个矩阵的第一行的第一个值，组成第一维的第一行，每一个矩阵的第一行第二个值组成第一维的第二行，以此类推，，每一个矩阵的第一行第n个值，组成第一维的第n行。
• 每个矩阵的第二行的第一个值，组成第二维的第一行，每一个矩阵的第二行第二个值组成第二维的第二行，以此类推，每一个矩阵的，第二行第n个值，组成第二维的第n行。
• 以此类推
• 每个矩阵的第n行的第一个值，组成第n维的第一行，每一个矩阵的第n行第二个值组成第n维的第二行，以此类推，每一个矩阵的，第n行第n个值，组成第n维的第n行。size=(n，y，i)