一维tensor:向量
- shape: torch.Size([10]) (10个元素举例)
- data
tensor([-0.1345, -0.6423, 1.0208, -0.9232, 1.1923, -1.1397, 1.4297, -0.5168,-2.6687, -0.6039])(注意这里是一个中括号)
二维tensor:矩阵
- shape:torch.Size([1,10])
- data:
tensor([[-1.2320, -0.4606, -0.4050, 1.2633, 0.3945, -0.0284, 0.8834, 0.7251, 1.1978, -1.1629]])(**注意这里是两个中括号)
三维tensor:RGB图像
- shape:torch.Size([3,10,10])
四维tensor:
Conv2d等卷积函数的输入tensor:分别是 batchsize,channel,高度,宽度。
*小技巧:通过数中括号的个数,就能得出数据的维度。