AttributeError: 'MNIST' object has no attribute 'shape'
时间: 2023-09-27 07:05:58 浏览: 82
Python3下错误AttributeError: ‘dict’ object has no attribute’iteritems‘的分析与解决
5星 · 资源好评率100%
这个错误通常是由于试图在MNIST对象上调用shape属性引起的,但是MNIST类中并没有shape属性。MNIST类是一个用于加载MNIST数据集的类,它包括训练数据、测试数据和标签。这个错误通常发生在试图对MNIST对象进行形状操作时,例如在使用某些机器学习模型时需要将输入数据转换为特定的形状。
如果您需要查看MNIST数据集的形状,可以使用以下代码:
```python
from torchvision import datasets
train_data = datasets.MNIST(root='data', train=True, download=True)
print(train_data.data.shape)
```
这将输出训练数据的形状,即 (60000, 28, 28)。如果您需要将MNIST数据集转换为其他形状,可以使用PyTorch中的torch.reshape()函数。例如,将MNIST数据集转换为形状 (60000, 784) 的张量,可以使用以下代码:
```python
train_data = datasets.MNIST(root='data', train=True, download=True)
train_data.data = train_data.data.reshape(-1, 784)
print(train_data.data.shape)
```
这会将MNIST数据集转换为形状为 (60000, 784) 的张量。
阅读全文