mindspore.nn.MultiheadAttention 2.7.2 文档 升思MindSpore社区
ms.float32) key=ms.Tensor(np.random.randn(seq_length,batch_size,embed_dim),ms.float32) value=ms.Tensor(np.random.randn(seq_length,batch_size,embed_dim),ms.float32) multihead_attn=ms.nn.MultiheadAttention(embed_dim,num_heads) attn_output,attn_output_weights=multihead_attn(query,key,value) print(attn_output.shape)(10, 8, 128)