diff --git a/beginner_source/basics/transforms_tutorial.py b/beginner_source/basics/transforms_tutorial.py index c6a6c5c1790..2c8515c872a 100644 --- a/beginner_source/basics/transforms_tutorial.py +++ b/beginner_source/basics/transforms_tutorial.py @@ -26,6 +26,7 @@ To make these transformations, we use ``ToTensor`` and ``Lambda``. """ +import torch from torchvision import datasets from torchvision.transforms import ToTensor, Lambda