pytorch - pytorch中张量torch.Size([])和torch.Size([1])的形状差异

Question

我是pytorch的新手。在玩张量时，我观察到了两种类型的张量-

tensor(58)
tensor([57.3895])

我打印了它们的形状，输出分别是 -

torch.Size([])
torch.Size([1])

两者有什么区别？

Answer 1

第一个有 0 大小维度，第二个有 1 维度，PyTorch 试图使两者兼容( 0 大小可以被视为类似于 float 或类似的，尽管我还没有真正遇到过明确需要它的情况，除了 @javadr 显示的内容在他下面的回答中)。
通常你会使用 list 来初始化它， see here 获取更多信息。