Pythorch:获取输入层大小

假设您的模型名为 model ,这将给出 fc1 图层:

model.fc1.in_features

这在 .forward() 方法:

def forward(self, x):
    x = x.view(-1, self.fc1.in_features)  # resize the input to match the input layer
    ...