Python 如何挤压除一个火炬暗淡以外的所有火炬？

可以将张量的形状转换为不具有大小为1的标注

我想在所有维度中压缩我的张量，只有一个维度除外（在本例中，不是压缩

dim=0

）

我在文件里看到的只有

dim（int，可选）–如果给定，输入将仅压缩在 这个维度

我想要相反的结果：

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

squeezed = torch.magic_squeeze(keep_dim=3)

assert squeezed == (5, 6, 1, 7)

---

可以这样做吗？

您可以使用

unsqueze（）

挤压并添加维度：

---

重塑将让您完成您想要做的事情：

import torch

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

---

谢谢，我在找更内置的东西：）

import torch

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