diff --git "a/chapter2-\345\277\253\351\200\237\345\205\245\351\227\250/chapter2: PyTorch\345\277\253\351\200\237\345\205\245\351\227\250.ipynb" "b/chapter2-\345\277\253\351\200\237\345\205\245\351\227\250/chapter2: PyTorch\345\277\253\351\200\237\345\205\245\351\227\250.ipynb"
index 19ffce43..47fbc1bc 100644
--- "a/chapter2-\345\277\253\351\200\237\345\205\245\351\227\250/chapter2: PyTorch\345\277\253\351\200\237\345\205\245\351\227\250.ipynb"	
+++ "b/chapter2-\345\277\253\351\200\237\345\205\245\351\227\250/chapter2: PyTorch\345\277\253\351\200\237\345\205\245\351\227\250.ipynb"	
@@ -1203,6 +1203,7 @@
    "outputs": [],
    "source": [
     "import torchvision as tv\n",
+     "import torch as t\n",
     "import torchvision.transforms as transforms\n",
     "from torchvision.transforms import ToPILImage\n",
     "show = ToPILImage() # 可以把Tensor转成Image，方便可视化"