为啥 torch.FloatTensor([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]]) 的大小是 [2,2, 3]？ [关闭]

Posted 2023-02-23

【中文标题】为啥 torch.FloatTensor([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]]) 的大小是 [2,2, 3]？ [关闭]

【英文标题】：Why torch.FloatTensor([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]]) size is [2,2,3]? [closed]为什么 torch.FloatTensor([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]]) 的大小是 [2,2, 3]？ [关闭]

【发布时间】：2020-06-16 13:36:42

【问题描述】：

>>> ft = torch.FloatTensor([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]])
>>> print(ft.shape)
torch.Size([2, 2, 3])

我无法理解这个结果。 我觉得torch size应该是[2,3,2]，结果却是[2,2,3]。

【问题讨论】：

为什么你认为它应该是别的东西？

【参考方案1】：

因为

len([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]]) = 2

这是前 2 个。 以及里面的每一项：

len([[0,1,2],[3,4,5]]) = 2

这是第二个2。

以及里面的每一项：

len([0,1,2]) = 3

这是第 3 个。

【讨论】：

哦，我以为尺寸是 [batch size, length, dim] 但事实并非如此。非常感谢。