torch.unsqueeze 发表于 2019-10-15 更新于 2020-05-07 阅读次数: 本文字数: 77 阅读时长 ≈ 1 分钟 torch.unsqueeze()的作用是增加一个维度,常用的是将单张图片增加一个维度。这样才可以传入模型。1torch.unsqueeze(img,0)