Pytorch：将[1，x]大小的张量转换为[x]大小的张量

Question

当尝试在Pytorch 0.4.0中加载可能是由Pytorch 0.3.1生成的模型时，我不断收到这样的错误：

While copying the parameter named "conv1_7x7_s2_bn.bias", whose dimensions in the
model are torch.Size([64]) and whose dimensions in the checkpoint are torch.Size([1, 64]).

我认为，如果我在每个张量上应用transpose，它将起作用，但是仍然失败，因为尺寸变为[64, 1]，而不是[64]，我需要。

我可以删除多余的维，从而将1行矩阵转换为向量吗？

注意：致电torch.flatten时，我得到：

AttributeError: module 'torch' has no attribute 'flatten'

Answer 1

删除空尺寸称为“挤压”。 NumPy does it，Tensorflow does it和PyTorch does it。

因此正确的命令是：

torch.squeeze(tensor)