weixin_40101756的博客

img = np.array(Image.open(''cat.png'')) # 加载图片并转为numpy格式inp = torch.from_numpy(img) 将numpy格式数据转为tensorpermute(2, 0, 1) # 将三维张量的维度进行变换，原始为0，1，2，变换后为2，0，1，可以理解我轴的变换。transpose(1,2) # 将维度1 ，2进行调换，因此permute用于多个维度调换，transpose用于两个调换unsqueeze() #...