我有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.