如何在pytorch中提取Alex网的最后一个隐藏层的特征向量?

时间:2017-10-13 09:45:21

标签: python neural-network pytorch

我在pytorch训练了一个Alex网络,我想从图层中提取特征向量 我应该使用什么功能?

1 个答案:

答案 0 :(得分:1)

在我看来,你可以在模型类中定义一个函数,它接受输入并输出你喜欢的功能,例如:

class model(nn.Module):
    def __init__(self):
        # init codes
    def forward(self, input):
        # forward codes
    def yourfunc(self, input):
        # codes
        #return feature1, feature2

yourfunc 仅采用您需要的输入和输出功能,不进行反向计算。您只需在任何需要的地方拨打电话即可。而且我不认为PyTorch中有内置函数可以做到这一点,因为它很容易自己实现。