模块级einsum解析
聚焦模块实现,适合动手写代码前逐层核对。
提示词
请以资深复现工程师身份,解析论文中的每个模块,输出模块级表格。要求:
(1) 名称与来源公式;
(2) 输入输出shape(用具体维度示例);
(3) 可学习参数及shape;
(4) 子操作顺序(按前向流程);
(5) 复现风险(如维度不匹配、初始化方式未知)。
所有公式必须用einsum表示,并校验张量维度一致性。缺失细节不脑补,标注“需推断”并进入风险登记表。示例输出
Attention模块:Q,K,V形状[B,L,d],einsum('bld,bmd->blm',Q,K)得到注意力分数,缩放因子需推断。
评论 0
更多
登录后可点赞、收藏、评论和举报。
还没有评论,先发起一个具体问题。