PyTorch中MultiheadAttention的输入参数是什么？

在PyTorch中，MultiheadAttention类有三个输入参数，分别为query，key和value。它们都是张量（tensor）形式的数据类型。其中，query用于计算注意力权重，key和value用于计算加权和，并输出一个张量表示注意力输出。这三个参数在MultiheadAttention中都有一个共同的维度大小，这是可配置的，通常可以设置为单个查询和键值向量的维度。通过指定这些参数，可以在深度学习模型的编程中实现多头注意力（Multi-head Attention）和其它一些任务。