Merge pull request apache#859 from u1234x1234/master

update data_iter example
kehao95 · Dec 14, 2015 · 4efaa42 · 4efaa42
2 parents 71a7128 + 8b81bb2
commit 4efaa42
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/example/python-howto/data_iter.py b/example/python-howto/data_iter.py
@@ -49,7 +49,11 @@
         round_batch=True)
 
 batchidx = 0
-for data, label in dataiter:
+for dbatch in dataiter:
+    data = dbatch.data[0]
+    label = dbatch.label[0]
+    pad = dbatch.pad
+    index = dbatch.index
     print("Batch", batchidx)
     print(label.asnumpy().flatten())
     batchidx += 1