torch.utils.data.Dataset类

Dataset是一个包装类,用来将数据包装为Dataset类,然后传入DataLoader中。他是一个重写类,可以通过继承和重写这个类就可以定义自己的数据类,需要重定义__len__(),和__getitem__()这两个函数。
len():返回数据的长度
Pytorch:torch.utils.data.Dataset类和torch.utils.data.DataLoader类
getitem():获取数据的方式
Pytorch:torch.utils.data.Dataset类和torch.utils.data.DataLoader类

torch.utils.data.DataLoader

Dataset,但是通过这种方式只能一个个的数据的把数据全部读出来,定义了数据读取的方式,不能实现** 批量**的把数据读取出来,为此pytorch有提供了一个方法:DataLoader()
Pytorch:torch.utils.data.Dataset类和torch.utils.data.DataLoader类

相关文章:

  • 2021-05-26
  • 2021-05-08
  • 2022-01-11
  • 2022-01-23
  • 2022-12-23
  • 2022-12-23
  • 2022-01-18
  • 2021-11-21
猜你喜欢
  • 2021-09-19
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案