Python中enumerate用法详解

代码分解

for idx, module in enumerate(args):

1. args 参数

• *args 表示函数可以接受任意数量的参数

• 在 __init__ 方法中，args 是一个包含所有传入参数的元组

2. enumerate(args) 函数

• enumerate() 是Python内置函数

• 它会遍历一个序列（如列表、元组），同时返回索引和对应的值

• 格式：(索引, 元素值)

3. 具体例子

假设这样使用：

model = MySequential(
    nn.Linear(10, 20),
    nn.ReLU(),
    nn.Linear(20, 5)
)

那么：

• args = (nn.Linear(10, 20), nn.ReLU(), nn.Linear(20, 5))

• enumerate(args) 会产生：

  • (0, nn.Linear(10, 20))

  • (1, nn.ReLU())

  • (2, nn.Linear(20, 5))

4. 循环过程

第一次循环：

• idx = 0, module = nn.Linear(10, 20)

• self._modules["0"] = nn.Linear(10, 20)

第二次循环：

• idx = 1, module = nn.ReLU()

• self._modules["1"] = nn.ReLU()

第三次循环：

• idx = 2, module = nn.Linear(20, 5)

• self._modules["2"] = nn.Linear(20, 5)

5. 最终结果

self._modules = {
    "0": nn.Linear(10, 20),
    "1": nn.ReLU(),
    "2": nn.Linear(20, 5)
}

简单理解

这句代码的作用就是：给每个传入的神经网络层自动编号并保存起来，方便后面按顺序执行。

这样设计让 MySequential 类可以接受任意数量的层，并且自动管理它们的执行顺序。