如何挤压除一个手电筒外的所有变暗？

如何解决如何挤压除一个手电筒外的所有变暗？

torch.squeeze 可以将张量的形状转换为不具有大小为 1 的维度。

我想在所有维度上压缩我的张量，但只有一个维度（在本例中，不是挤压 dim=0）。

我在文档中只能看到

dim (int,optional) – 如果给定，输入只会被压缩 这个维度

我想要相反的：

t = torch.zeros(5,1,6,7,1)

squeezed = torch.magic_squeeze(keep_dim=3)

assert squeezed == (5,7)

这能做到吗？

解决方法

重塑会让你完成你想做的事：

import torch

t = torch.zeros(5,1,6,7,1)
t = t.reshape((5,7))
>>> torch.Size([5,7])

,

您可以使用 unsqueeze() 挤压并添加维度：

import torch

t = torch.zeros(5,1)
squeezed = t.squeeze().unsqueeze(2)
print(squeezed.shape)
>>> torch.Size([5,7])