pytorch 模型去掉最后一层 或 最后几层

pytorch的模型去掉末端的n层，以n=1为例：

new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )

ps: 模型如何去掉任意位置的某一层？好像不支持？可能去掉之后，图就改变了，问题比较复杂。但是可以通过在load statedict时选择性的加载有需要的参数层，来间接实现这一目的。