有关python的iter,next,yield,和pytorch的dataloader
问题首先,对于一个python数组,我们可以用for,或者next来遍历。其次,而对于pytorch的 torch.utils.data.DataLoader, 也可以使用类似 next(iter(DataLoader)) 的方式遍历地读取dataset的数据。再次,使用yield关键字,也可以起到“遍历”的效果。那么问题来了。这iter,next,yield几个东西之间有什么关联,又有什么区别呢?pytorch的DataLoader又是用的什么样的方式呢?分析先看一个简单例子:fruit =
原创
2021-12-08 13:29:14 ·
7723 阅读 ·
2 评论