torch.stack()中的dim

torch.stack()中的dim

浏览了一篇关于torch.stack()的博客，记录自己对dim的理解。
举个例子：
A和B是个m*n的tensor，C=torch.stack((A,B),dim=i)
i=0时，C[0]=A,C[1]=B，A和B没有任何维度上的堆叠；
i=1时，C[0]=[A[0],B[0]],C[1]=[A[1],B[1]]······，相当于A和B在行这个维度上堆叠；
i=2时，C[0]=[ [A[0][0],B[0][0]],······,[A[0][n-1],B[0][n-1]] ]······，相当于A和B在列这个维度上的堆叠。