获取DataLoader中的数据
DataLoader返回的是一个迭代器,其中每次迭代返回一个数组(数组中的元素是迭代器中返回的内容NdArray)。例如
train, test = d2l.load_data_fashion_mnist(batch_size=1)
#train是一个DataLoder,含有features,labels(他们都是NDArray类型的)
for features, labels in train:
print(features, labels)
break;
#features,labels(他们都是NDArray类型的)
for zips in train:
print(type(zips))
print(zips)
break
'''
输出:
list
[
fratures
<NDArray>
labels
<NDArray>
]
输出的是将两个返回值放入数组中的数据
'''
获取DataLoader中的全部数据:
train, test = d2l.load_data_fashion_mnist(batch_size=1)
i = 0
for f, l in train:
if i==0:
features = f
continue
features = nd.concat(features, f, dims = 0)
#只需要第一次弄一个与f形状相同的NDArray然后就能一直concat了,最后的features就是数据了