python - 删除pytorch中的零维

Question

我有尺寸为 的 torch.tensor 0 x 240 x 3 x 540 x 960
我只想删除 0 维，所以我想要 pytorch 张量
的尺寸240 x 3 x 540 x 960 .
我用过 tensor= torch.squeeze(tensor)尝试这样做，但没有删除零维...
就我而言，张量的大小是可变的，因此我无法将其硬编码为 torch.squeeze..
有没有什么简单的方法可以删除 pytorch.tensor 中不需要的维度？

Answer 1

你不能。
形状为 0 x 240 x 3 x 540 x 960 的张量的大小(= 元素数)是 0。
您无法将其 reshape 为形状张量 240 x 3 x 540 x 960因为它有 373248000 个元素。
squeeze() 文档:

Returns a tensor with all the dimensions of input of size 1 removed.

这将删除 1 大小的暗淡，而不是 0 大小的暗淡，这是有道理的。

无论如何，原始张量中的数据量为 0，它可以表示为任何至少具有一个零暗度的形状，但不能表示为大小 > 0 的形状。