Transformer中的Attention怎么能够打印出来？

想要看看Transformer中Attention的情况，怎么能够打印出来让我进行查看呀？

1. 自己继承库内的类，然后修改进行输出。
2. 直接自己写一个Transform类进行debug。
3. 一般都是看transform输出，你看内部也没啥东西啊？