axes don't match array in pytorch

一个有意思的错误: self.transformer = transforms.Compose([transforms.ToTensor()]) img = self.transformer(img) 该语句在python3.7中运行无问题,在python2.7中出现如下错误: img = torch.from_numpy(pic.transpose((2, 0, 1))) ValueError:
相关文章
相关标签/搜索