参考:
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]])
解释: