Refer :https://blog.csdn.net/shenfuli/article/details/106523650

Multi-Head Attention: https://blog.csdn.net/qq_37394634/article/details/102679096