深度学习模型输出特征矩阵

有没有大佬知道:训练好的transformer模型,我把测试样本放进去后怎么在全连接层前输出特征矩阵

你的输出是指什么,网络输出口还是指打印输出?
如果是print的话直接在model的全连接层之前print一下就行,如果是网络输出口的话,有两种方法,一种是定义一个输出口,另外一种就是在在forward里面return model的时候多return一个特征矩阵。
类似于这样

def forward(self,x):
    layer1=net1(x)
    layer2=net2(layer1)
    ... 
    layer_n=net_n(layer_n-1)
    feature_map=layer_n
    fc1=fc(layer_n)
    return fc1,feature_map