NameError：名称“transforms”未定义，而使用 google colab

Question

train = datasets.MNIST("", train=True, download=True,transform = transforms.Compose([transforms.ToTensor()]))
test = datasets.MNIST("", train=False, download=True,transform = transforms.Compose([transforms.ToTensor()]))`

在 colab notebook 上执行此操作后，出现此错误：

Traceback (most recent call last)
<ipython-input-6-b81aa6cf1cbe> in <module>()
----> 1 train = datasets.MNIST("", train=True, download=True,transform = transforms.Compose([transforms.ToTensor()]))
      2 test = datasets.MNIST("", train=False, download=True,transform = transforms.Compose([transforms.ToTensor()]))

NameError: name 'transforms' is not defined*

Answer 1

我从上下文中猜测您正在使用 Pytorch，在这种情况下，您需要确保：

from torchvision import transforms

在你的进口。从你已经datasets从同一个库中导入的东西来看。