代码参考的是：https://blog.csdn.net/qq_38668236/article/details/126503809训练可以正常执行 但是验证的时候就开始报错
之前加过其他注意力模块都没报错 请各位指点指点
