交叉注意力机制

调用nn.MultiheadAttention(embed_dim,num_heads=1)可以实现

输入q,k,v，另外设置need_weights参数可以控制是否输出权重

输出元组。结果是元组第0个元素，它的维度是(L,N,embed_dim)，L是序列步长，它的值取决于q的步长。