5.1.4 量子注意力机制的代码实现