dym755833564的博客

1. squeeze()压缩张量，去除size为1的维度2.unsqueeze（）增加维度：在指定维度上增加维度，如b = torch.tensor([1, 2, 3]), torch.Size([3])在dim = 0的维度插入，B1 = torch.unsqueeze(b, 0)，输出为[[1,2,3]]。torch.Size([1, 3])加到另一个维度，B2 = B.unsqueeze(1)，输出为[[1],[2],[3]]。torch.Size([3, 1])在B1基础

import torchimport torchvision.models as modelsimport MyDataLoaderfrom torch.autograd import Variablealexnet = models.alexnet(pretrained = False)alexnet.load_state_dict(torch.load('/home/alex...

报错如下：RuntimeError: Found 0 files in subfolders of: /home/A/0Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,webp产生原因：路径问题举例说明： /home/A/0/ 目录下存在若干图片，/home/A/ 目录下只有 0 这一个文件夹。那...