How to extratct the attention weight of the source to target text

nlpyang / PreSumm

code for EMNLP 2019 paper Text Summarization with Pretrained Encoders

MIT License

1.29k stars 465 forks source link

Closed teejay13 closed 3 years ago

teejay13 commented 3 years ago

I want to generate the attention weight of the source to the target text in order to visualize the distribution. Is that possible