ó ùµÈ[c@@sVdZddlmZddlmZmZddlmZdefd„ƒYZ dS( s/Contrib data iterators for common data formats.i(tabsolute_importi(tDataItertDataDesc(tndarraytDataLoaderItercB@sVeZdZdddd„Zd„Zd„Zd„Zd„Zd „Zd „Z RS( sSReturns an iterator for ``mx.gluon.data.Dataloader`` so gluon dataloader can be used in symbolic module. Parameters ---------- loader : mxnet.gluon.data.Dataloader Gluon dataloader instance data_name : str, optional The data name. label_name : str, optional The label name. dtype : str, optional The dtype specifier, can be float32 or float16 Example usage: ---------- >>> import mxnet as mx >>> from mxnet.gluon.data.vision import MNIST >>> from mxnet.gluon.data import DataLoader >>> train_dataset = MNIST(train=True) >>> train_data = mx.gluon.data.DataLoader(train_dataset, 32, shuffle=True, num_workers=4) >>> dataiter = mx.io.DataloaderIter(train_data) >>> for batch in dataiter: ... batch.data[0].shape ... (32L, 28L, 28L, 1L) tdatat softmax_labeltfloat32cC@sİtt|ƒjƒ||_t|jƒ|_t|jƒ\}}|jd|_||_ t ||j|ƒg|_ t ||j|ƒg|_ d|_|jƒdS(Ni(tsuperRt__init__t_loadertitert_itertnexttshapet batch_sizetdtypeRt provide_datat provide_labeltNonet_current_batchtreset(tselftloadert data_namet label_nameRRtlabel((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyR 5s   cC@st|jƒ|_dS(N(R R R (R((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyRAscC@s@yt|jƒ|_Wntk r2d|_nX|jdk S(N(R R Rt StopIterationR(R((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyt iter_nextDs   cC@s„|jƒrj|jdj}tjd|jgt|dƒƒ}|jdj|jƒ||d*|gS|jdj|jƒgS(NiRi( tgetpadRRtndtemptyRtlisttastypeR(Rtdshapetret((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pytgetdataKs  &!cC@s„|jƒrj|jdj}tjd|jgt|dƒƒ}|jdj|jƒ||d*|gS|jdj|jƒgS(NiRi( RRRRRRR R!R(RtlshapeR#((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pytgetlabelSs  &!cC@s|j|jdjdS(Ni(RRR(R((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyR[scC@sdS(N(R(R((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pytgetindex^s( t__name__t __module__t__doc__R RRR$R&RR'(((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyRs     N( R*t __future__RtioRRtRRR(((sP/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/io.pyts