题目描述

给定 Query 序列

和 Key/Value 序列

请你实现交叉注意力（Cross Attention）的完整计算流程。

1. 线性变换

2. 计算注意力分数

3. 对每个 Query 行进行 softmax

4. 得到加权输出

5. 输出层线性变换

输入参数

• X_Q：形状为 $T_Q \times d_{\text{model}}$ 的 Query 序列
• X_K：形状为 $T_K \times d_{\text{model}}$ 的 Key/Value 序列
• W_Q, W_K, W_V：形状为 $d_{\text{model}} \times d_{\text{model}}$ 的线性投影矩阵
• b_Q, b_K, b_V：长度为 $d_{\text{model}}$ 的偏置向量
• W_O：形状为 $d_{\text{model}} \times d_{\text{model}}$ 的输出层权重
• b_O：长度为 $d_{\text{model}}$ 的输出偏置

返回值

• O：形状为 $T_Q \times d_{\text{model}}$ 的最终交叉注意力输出矩阵

示例

输入：

X_Q =
[[1, 0, 1, 0],
 [0, 1, 0, 1]]

X_K =
[[1, 1, 0, 0],
 [0, 1, 1, 0]]

W_Q =
[[1, 0, 1, 0],
 [0, 1, 0, 1],
 [1, 0, 0, 1],
 [0, 1, 1, 0]]

b_Q = [0, 0, 0, 0]

W_K =
[[1, 1, 0, 0],
 [0, 1, 1, 0],
 [1, 0, 0, 1],
 [0, 0, 1, 1]]

b_K = [0, 0, 0, 0]

W_V =
[[1, 0, 0, 1],
 [1, 1, 0, 0],
 [0, 1, 1, 0],
 [0, 0, 1, 1]]

b_V = [0, 0, 0, 0]

W_O =
[[1, 0, 1, 0],
 [0, 1, 0, 1],
 [1, 0, 0, 1],
 [0, 1, 1, 0]]

b_O = [0, 0, 0, 0]

输出：

O =
[[2.00, 2.00, 1.76, 2.24],
 [2.00, 2.00, 2.24, 1.76]]

提示

• 输入序列范围： $-1000 \le X_Q[i,j],\ X_K[i,j] \le 1000$
• 权重矩阵范围： $-10 \le W_Q,\ W_K,\ W_V,\ W_O \le 10$
• 偏置项范围： $-5 \le b_Q,\ b_K,\ b_V,\ b_O \le 5$
• softmax 输出满足： $0 \le A[i,j] \le 1$ $\sum_{j} A[i,j] = 1$
• 输出 $O$ 可为任意实数矩阵