Python3中.pyi文件介绍

最新推荐文章于 2024-06-13 11:38:29 发布

fengbingchun

最新推荐文章于 2024-06-13 11:38:29 发布

阅读量2.2w

点赞数 9

分类专栏： Python 文章标签： stub file

本文链接：https://blog.csdn.net/fengbingchun/article/details/125302769

版权

Python 专栏收录该内容

61 篇文章

订阅专栏

PyTorch中的.pyi文件是存根文件，用于提供模块的类型信息和公共接口描述，不含实际运行时代码。这些文件在Python3中用于类型提示和静态类型检查，帮助开发者在编写代码时获得更好的类型检查支持。每个优化算法的.py文件都有对应的.pyi文件，例如SGD的.pyi文件仅包含__init__函数的类型定义。创建存根文件可以手动编写或通过工具，且需设置MYPYPATH环境变量。在一些IDE如WingPython和PyCharm中，.pyi文件的设置有助于提升开发体验。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在看PyTorch代码时，在一些模块中经常会发现.pyi文件，如下图所示，是PyTorch中torch/optim模块中的.pyi文件：

每种实现优化算法的.py文件，都有一个对应的.pyi文件，每个.pyi文件中的内容都相似，仅有类的__init__函数，如SGD优化算法的sgd.pyi内容如下：

from .optimizer import _params_t, Optimizer

class SGD(Optimizer):
    def __init__(self, params: _params_t, lr: float, momentum: float=..., dampening: float=..., weight_decay:float=..., nesterov:bool=...) -> None: ...

在Python3中，.pyi文件是存根文件(stub file)。这个"pyi"中的"i"代表接口即interface，作为公共接口。在Python中不存在"interface"，只是提供一个特殊的扩展。它是一个普通的Python文件，除了文件中的所有的方法都是空的。

存根文件仅包含模块公共接口的描述，没有任何实现。

.pyi文件中允许使用"..."代替任何实现细节。它是仅包含类型信息的文件，没有运行时代码。

.pyi文件是一个Python skeleton，具有适当的结构、调用签名和返回值，以匹配模块中定义的函数、属性、类和方法。

创建存根文件：

(1).为模块编写一个存根文件，并将其作为.pyi文件存储在与模块相同的目录中；

(2).或者将.pyi文件单独存放在为存根保留的目录中，在这种情况下，需要设置环境变量MYPYPATH。

Wing Python IDE和PyCharm中会涉及到对.pyi文件的设置。

GitHub：https://github.com/fengbingchun/PyTorch_Test