简单Transformer

追加 KV Cache

请按照下方说明实现 kv_append()。

将新的 key 和 value 向量追加到已有 KV Cache。Cache 由两个 token 向量列表组成：k_cache（key 向量列表）和 v_cache（value 向量列表）。给定一个新的 key 向量和 value 向量，返回追加新向量后的（k_cache, v_cache）元组。

每个 key/value 向量都是长度相同的浮点数列表。

检查项正确性形状数据类型

空 Cache

正确性

调用

kv_append([], [], [1, 2], [3, 4])

预期结果

[[[1, 2]], [[3, 4]]]

已有条目

正确性

调用

kv_append([[0, 0]], [[0, 0]], [1, 1], [2, 2])

预期结果

[[[0, 0], [1, 1]], [[0, 0], [2, 2]]]

Python 3 · Python · 函数题 · kv_append()

运行会检查可见示例，提交会运行完整的隐藏测试。

登录并提交后，你的提交记录会显示在这里。