Pytorch之张量维度及不同维度的操作

参考：

pytorch 矩阵维度 - 搜索结果 - 知乎

Pytorch 中的 dim操作介绍 - 大数据 - 亿速云

1.如何理解dim?

• pytorch的dim和numpy的axis很类似
• 不同dim的数据长什么样?

维度为0, 0维张量也叫标量     1                                                             
维度为1, 0维张量也叫矢量   [1,2]                      
维度为2, 0维张量也叫矩阵    [[1,2],[3,4]]   
维度为3, 0维张量也叫矩阵数组 [[[1,2],[3,4]],[[1,2],[3,4]]]     

二维矩阵a:

a = torch.tensor([[1, 2], [3, 4]])
print(a)
tensor([[1, 2],
        [3, 4]])

解释：