imgs = loader_train.__iter__().next()[0].view(batch_size, 784).numpy(),怎么修改
时间: 2024-04-15 10:27:38 浏览: 145
练习三.zip
要修改这行代码,你可以使用以下方式:
```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数组。
阅读全文