我是靠谱客的博主 称心大侠,最近开发中收集的这篇文章主要介绍yield加载数据1. yield生成器,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

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生成器所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(31)

评论列表共有 0 条评论

立即
投稿
返回
顶部