上述sep_attn函数如何实现

sep_attn函数的实现如下：

1. 首先，对输入的三个张量进行维度检查，确保它们的维度都正确。

2. 接着，将输入的三个张量进行线性变换，得到query、key和value张量。

3. 实现scaled dot-product attention。首先，将query和key张量相乘，得到一个张量，然后将其除以根号下d，其中d是query和key张量的维度。接着，将结果通过softmax函数进行归一化，得到一个张量。最后，将归一化的张量与value张量相乘，得到最终的输出张量。

4. 返回输出张量。