删除pytorch中的零尺寸



我有torch张量,其尺寸为0 x 240 x 3 x 540 x 960

我只想删除0维,所以我想有pytorch张量

尺寸240 x 3 x 540 x 960

我用tensor= torch.squeeze(tensor)来尝试,但零维度没有被删除。。。

在我的情况下,张量的大小是可变的,所以我不能把它硬编码为torch

有没有什么简单的方法可以去除pytorch.ttensor中不需要的维度?

你不能

形状为0 x 240 x 3 x 540 x 960的张量中的大小(=元素的数量(是0。

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

squeeze()文档:

返回一个张量,其中删除了大小为1的输入的所有维度。

这会删除1大小的调光,而不是0大小的调暗,这是有意义的。


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

最新更新