Python mxnet.contrib.io.DataLoaderIter用法及代码示例

用法: class mxnet.contrib.io.DataLoaderIter(loader, data_name='data', label_name='softmax_label', dtype='float32')

参数：

loader：(mxnet.gluon.data.Dataloader) - Gluon 数据加载器实例
data_name：(str, optional) - 数据名称。
label_name：(str, optional) - 标签名称。
dtype：(str, optional) - dtype 说明符，可以是 float32 或 float16

基础：mxnet.io.io.DataIter

返回 mx.gluon.data.Dataloader 的迭代器，因此可以在符号模块中使用胶子数据加载器。

例子：

>>> 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)

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.contrib.io.DataLoaderIter。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。