在Latex中插入python代码

在Latex中插入Python代码，需要一个第三发的宏包pythonhighlight:

https://github.com/olivierverdier/python-latex-highlighting

下载pythonhighlight.sty后，将它放到你的.tex文件所在目录下。

然后声明要使用pythonhighlight，在tex文件内的导言区：

\usepackage{graphicx}
\usepackage{pythonhighlight}

之后既可以在正文添加代码了:

\begin{python}
import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
import torchvision
import torchvision.transforms as transforms
import time
import os
\end{python}

效果如下：