- 标量的形状是
[]
- 向量的形状是
[N]
,N
是向量元素的个数 - 二维表格的形状是
(M,N)
,M
是行数,N
是列数 - 三维矩阵的形状是
(M,N,C)
,M
是长,N
是宽,C
是高
>>>
>>> import torch as th
>>>
>>> th.tensor(2)
tensor(2)
>>> th.tensor([2])
tensor([2])
>>> th.tensor([[2]])
tensor([[2]])
>>> th.tensor([[[2]]])
tensor([[[2]]])
>>>
>>> th.tensor(2).shape
torch.Size([])
>>> th.tensor([2]).shape
torch.Size([1])
>>> th.tensor([[2]]).shape
torch.Size([1, 1])
>>> th.tensor([[[2]]]).shape
torch.Size([1, 1, 1])
>>>
>>>