the_continent的博客

原创 初学torch.stack()对dim的个人理解

最近在看这里的内容，对于“维度”、“折叠”这些表述很难理解。在看了几篇讲解之后，目前自己形成了比较直观的理解。记录一下防止未来再产生同样的困惑。在这里我们定义两个2*3的张量a=[ [ 0 ,0 ,0 ] , [0 , 0 , 0] ]b=[ [ 1, 1, 1] , [1 , 1, 1]其实大于二维也一样，但是二维比较容易以传统矩阵的形式理解。我们把a b 写成矩阵的形式。下面就可以讲维度的问题了，对于a b 他们的第一维，第二维自然就是他们的行和列。stack函数其实就.