torch.size()
查看tensor的尺寸;
- 1-6行:两个
[]
两个数字,第一层[]
中以,
分割包含两个[]
,所以2;其中每一个[]
中又有3个数,所以torch.Size
的第二个数是三。 - 8-9行:在1-6基础上再加一个
[]
,所以第一个数字是1。
torch.size()
查看tensor的尺寸;
[]
两个数字,第一层[]
中以,
分割包含两个[]
,所以2;其中每一个[]
中又有3个数,所以torch.Size
的第二个数是三。[]
,所以第一个数字是1。