How to visualize the attention heatmap for textual entailment

Hello!

I am working on textual entailment. Can anyone please help on how to visualize the premise and hypothesis of activated neurons? similarly like this:

image

Best Regards

Did you end up finding a solution for this at all @anis016 ?