imgs = loader_train.__iter__().next()[0].view(batch_size, 784).numpy(),怎么修改
时间: 2024-04-15 21:27:38 浏览: 154
imgs.zip_imgs.com_imgs图片_web 图片展示_展示_左右滑动模
要修改这行代码,你可以使用以下方式:
```python
imgs = next(iter(loader_train))[0].view(batch_size, 784).numpy()
```
这里使用`next(iter(loader_train))`来替代`loader_train.__iter__().next()`。这两种方法是等效的,都是获取`loader_train`中的下一个batch数据。然后使用`.view(batch_size, 784)`来改变数据的形状,并最后使用`.numpy()`将数据转换为NumPy数组。
阅读全文