移除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.squence

有没有简单的方法可以去除pytorch.tensor中不需要的维度

Answer 1

你不能

形状0 x 240 x 3 x 540 x 960的张量中的大小（=元素数）为0

不能将其重塑为240 x 3 x 540 x 960形状的张量，因为它有373248000个元素

^{}文件：

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

这将删除1个大小的DIM，而不是0个大小的DIM，这是有意义的

---

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