lizezhonglaile/bert-attention-head-vis

Attention matrix visualization example

Jupyter Notebook