题目描述
实现旋转位置编码 RoPE。给定一个 query 向量 q(维度 d,偶数)和位置 pos,对每一对相邻维度
要求
- 实现右侧代码模板中给出的函数,签名以模板为准
实现旋转位置编码 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
还没有评论,来做第一个发言的人吧
运行结果
点击「调试」(只跑样例)或「提交」(跑全部用例)查看结果