dogdogpp/bert_hidden_staes

转onnx模型时导出bert模型的中间层输出，如“hiddens_states", "last_hidden_state”, "attentions"

Python