num_heads!=1时，attention计算报错

[rye]

如果将num_heads设为不等于1的值，在spatial-temporal/difformer.py中的第44行计算attention时，代码报错。