如何把注意力权重可视化

有没有同学知道nlp里面如何把一段话显示出哪些部分是被重点学习的？也就是注意力权重的可视化？
论文里想要展示出来，谢谢大家