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)