返回题库
#2中等

实现 Rotary Position Embedding (RoPE)

位置编码Attention

题目描述

实现旋转位置编码 RoPE。给定一个 query 向量 q(维度 d,偶数)和位置 pos,对每一对相邻维度 施加旋转:

要求

  • 实现右侧代码模板中给出的函数,签名以模板为准

示例

示例 1

输入:
q = [1.0, 0.0], d = 2, pos = 0
输出:
[1.0, 0.0]

说明:pos=0 时 theta=0,旋转矩阵为单位阵

示例 2

输入:
q = [1.0, 0.0, 1.0, 0.0], d = 4, pos = 1
输出:
[0.5403, 0.8415, 0.9999, 0.0100]

说明:第一对 theta=1,第二对 theta=0.01

讨论区(0)

还没有评论,来做第一个发言的人吧

Python · solution.py
编辑器加载中…

运行结果

点击「调试」(只跑样例)或「提交」(跑全部用例)查看结果