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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。