如何從 PyTorch model 的 `forward()` function 的源代碼中提取層？

Question

我想提取處理forward function 中定義的輸入數據的 model 的層。 例如，給定以下源代碼：

 def forward(self, inputs): 
     inputs = self.embedding(inputs)
     inputs = F.dropout(inputs, 0.25, self.training)
     return inputs

我想提取處理輸入數據的層，即：

input -> embedding -> dropout -> output

如何在不運行代碼的情況下執行此操作？

Answer 1

model.children()將遍歷網絡中的子模塊（層）。