首页 > 其他 > 详细

pytorch 从Dataset类中获取数据

时间:2021-05-27 15:13:16      阅读:33      评论:0      收藏:0      [点我收藏+]

转自:https://www.jianshu.com/p/4818a1a4b5bd

1.介绍

Dataset类是为torch.utils.data.DataLoader做准备,支持两种类型的访问

* map-style datasets #__getitem__()
* iterable-style datasets  #__iter__()
(1) 
print("trainDataset 的类型:", type(trainDataset))
>>> trainDataset 的类型: <class torchvision.datasets.mnist.MNIST>

(2)
print("trainDataset 的长度:", len(trainDataset))
>>> trainDataset 的长度: 60000

(3)
print("trainDataset[0] 的类型:", type(trainDataset[0]))
print("trainDataset[0] 的长度:", len(trainDataset[0]))
>>>  trainDataset[0] 的类型: <class tuple>
     trainDataset[0] 的长度: 2

(4)
print("trainDataset[0][0] 的类型:", type(trainDataset[0][0]))
print("trainDataset[0][0] 的形状:", trainDataset[0][0].shape)
>>>  trainDataset[0][0] 的类型: <class torch.Tensor>
     trainDataset[0][0] 的形状: torch.Size([1, 28, 28])

(5)
print("trainDataset[0][1] 的类型:", type(trainDataset[0][1]))
print("trainDataset[0][1] :", trainDataset[0][1])
>>>  trainDataset[0][1] 的类型: <class int>
     trainDataset[0][1] : 5

从上述代码可以看到,能够通过一些方法去访问。

pytorch 从Dataset类中获取数据

原文:https://www.cnblogs.com/BlueBlueSea/p/14817370.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!