一維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等卷積函數(shù)的輸入tensor:分別是 batchsize,channel,高度,寬度。
*小技巧:通過數(shù)中括號的個數(shù),就能得出數(shù)據(jù)的維度。
tensor對象屬性

圖片.png