Dataloader返回LoDTensor
收藏
在2.1版本中,按照API文件描述,paddle.io.DataLoader返回的是Tensor类型的变量,实际运行中,返回的是LoDTensor类型,代码及运行结果如下,恳请指教:
train_data = paddle.text.datasets.UCIHousing(mode='train')
train_loader = paddle.io.DataLoader(train_data, batch_size=batch_size, shuffle=True)
for data, label in train_loader():
print(type(data))
运行耗时: 28毫秒
0
收藏
请登录后评论
重启notebook之后运行正常……