pytorch如何定义变量