【pytorch】对numpy对象进行增加维度

使用unsqueeze()来增加维度：x = torch.from_numpy(img).float().unsqueeze(0)，其中的参数0是指“在第0个维度增加一维”

如果想再增加一个维度，则只需要再添加一行unsqueeze(0)即可