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.内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复