zoukankan      html  css  js  c++  java
  • torch.utils.data.DataLoader对象中的迭代操作

    关于迭代器等概念参考:https://www.cnblogs.com/zf-blog/p/10613533.html

    关于pytorch中的DataLoader类参考:https://blog.csdn.net/u014380165/article/details/79058479(写的蛮好)

    现在用一个例子来讲解一下:

    (1)法一:

    a = torch.utils.data.DataLoader(....)

    a = enumerate(a) #enumerate的built-in函数:__iter__()会将a变成iterator,后面可以再调用built-in函数:__next__()进行迭代;

    m = a.__next__()  #注意这里使用的是enumerate类型的built-in函数:__next__()

    (2)法二:

    a = torch.utils.data.DataLoader(...)

    a = torch.utils.data.DataLoader.__iter__(a) #这里调用DataLoader的built-in函数:__iter__()将a变成iterator;

    m = a.__next__() #注意:这里调用的是DataLoader的built-in函数:__next__()

    PS:

    (1)上述两种__next__()得到的封装结构是不同的,这个在拆解a时要注意;

    (2)将容器(各种数据结构:list、tuple等)变成iterator时需要使用iter(),而对于类操作时需要在类中定义__init__()和__next__(),注意这点区别;(具体可参考:https://blog.csdn.net/admin_maxin/article/details/82052065 , 这样也说明了为什么不能用a = iter() , m=a.next()这种做法,因为a是一个类了,需要用built-in函数__iter__将其变成iterator,再用__next__(),否则会报“enumarate” object has no attribute "next"的错误 or "DataLoader" object has no attribute "next")

  • 相关阅读:
    Python 面向对象
    Python __str__()
    数据降维
    高并发相关概念
    centos7下安装kubernetes1.18
    OB-运行日志
    OB-租户(Tenant)管理
    OB-资源管理(Resource Unit/Pool)
    [转载]-基于 VMWARE Oracle Linux7.9 安装 Oracle19c RAC 详细配置方案
    OB-管理oceanbase集群参数
  • 原文地址:https://www.cnblogs.com/zf-blog/p/10613846.html
Copyright © 2011-2022 走看看