nn.MultiHeadAttention的函数参数是什么？

`nn.MultiHeadAttention` 是 PyTorch 中的一个模块，用于实现多头注意力机制。它的参数包括：

- `embed_dim`：输入和输出张量的特征维度。
- `num_heads`：多头注意力的头数。
- `dropout`：可选的 dropout 比率，用于在计算注意力权重时进行随机失活。
- `bias`：是否使用偏置。
- `add_bias_kv`：是否为键值对计算注意力分数时添加偏置。
- `add_zero_attn`：是否在注意力分数张量的最后一维添加一个零向量。

在使用 `nn.MultiHeadAttention` 时，需要将输入张量和一个可选的键值对输入到模块中，并调用 `forward` 方法来计算输出张量。具体使用方法可以参考 PyTorch 的官方文档。