pycharm不能自动补全某些函数

在使用pycharm过程中，我发现很多pytorch的函数不能自动补全，也无法查看相应的源码，如torch.nn等。解决办法：

在pycharm中，点击File -> Settings -> Editor -> File Types，如下图，在Registered Patterns中增加*.pyi即可。