PyTorch 代码自动补全工具

安装完PyTorch后,运行程序输出正常,但是发现PyCharm有的函数却不会自动补全。如何解决? 方法就是写一个.pyi文件。这个文件里把你的函数、类的输入输出等定义好,PyCharm会自动识别。而在Python运行时又会直接忽略这个文件。因此我们就可以在完全不影响PyTorch本身的情况下为其增加类型提示。