我在pytorch训练了一个Alex网络,我想从图层中提取特征向量 我应该使用什么功能?
答案 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中有内置函数可以做到这一点,因为它很容易自己实现。