PyTorch之torch.utils.data
torch.utils.data主要包括下面三个类: 1.class torch.utils.data.Dataset 作用:创建数据集,有__getitem__(self, index)函数来根据索引序号获取图片和标签,有__len__(self)函数来获取数据集的长度。 其它的数据集类必须是torch.utils.data.Dataset的子类,比如说torchvision...
原创
2019-03-08 16:28:28 ·
996 阅读 ·
0 评论