问题 :
torch.Size([196608])我是不是可以理解成 它的原始形状是不是torch.Size([1, 1, 1, 196608],然后因为都是一维的,所以1可以不写,这样理解可以吗?
博主答案:
是的,你可以将 torch.Size([196608])
理解为原始形状为 torch.Size([1, 1, 1, 196608])
的一维张量。在这种情况下,由于该张量在所有维度上都是长度为 1,因此这些维度可以被省略,只保留一维的大小为 196608。
在 PyTorch 中,形状 (1, 1, 1, 196608)
和 (196608,)
都表示相同的一维张量。PyTorch 在显示张量形状时,默认会去除只有一个元素的维度,因此可以将其简化为 (196608,)
。
因此,你可以将 torch.Size([196608])
视为 (1, 1, 1, 196608)
的简化形式,其中的 1 维度被省略了。这种方式的理解是合理的,能够帮助你更好地理解张量的形状。