如何按索引访问PyTorch模型参数

如果我的网络有10层，包括偏置项，如何仅通过索引访问第i层的参数？

目前，我正在做类似这样的事情：

for parameter in myModel.parameters():    parameter.data /= 5

如何通过索引访问parameter.data？例如，我想在不迭代的情况下访问第9层的参数，类似于myModel.parameter.data[8]或类似的东西。

回答：

只需执行以下操作：

list(myModel.parameters())

现在它将成为一个权重和偏置的列表，要访问第一层的权重，可以这样做：

print(layers[0])

要访问第一层的偏置，可以这样做：

print(layers[1])

依此类推。

请记住，如果某一特定层的偏置设置为false，它将根本没有条目，因此例如，如果第二层的偏置为false，那么layers[3]实际上会给出第三层的权重。

学技术