概述
yield加载数据
- 1. yield生成器
1. yield生成器
和return后函数会结束不一样,Python的yieid可以实现多次return,因此非常适合用于深度学习数据加载。
class DataLoader(object):
def __init__(self):
self.n = 10
def feed(self):
for i in range(self.n):
yield i
if __name__ == '__main__':
dataLoader = DataLoader()
dataItem = dataLoader.feed()
for i in range(dataLoader.n):
print(next(dataItem))
- 实现生成器
DataLoader类负责读取数据,假如它读了10条数据。它的一个函数feed(函数名随意)使用yield产生数据。 - 使用生成器
调用dataLoader的feed函数获得返回值dataItem(第10行)。在循环中使用next函数获取生成内容(第12行)。
最后
以上就是称心大侠为你收集整理的yield加载数据1. yield生成器的全部内容,希望文章能够帮你解决yield加载数据1. yield生成器所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复