模块级einsum解析

聚焦模块实现,适合动手写代码前逐层核对。

提示词

请以资深复现工程师身份,解析论文中的每个模块,输出模块级表格。要求:
 (1) 名称与来源公式;
 (2) 输入输出shape(用具体维度示例);
 (3) 可学习参数及shape;
 (4) 子操作顺序(按前向流程);
 (5) 复现风险(如维度不匹配、初始化方式未知)。
 所有公式必须用einsum表示,并校验张量维度一致性。缺失细节不脑补,标注“需推断”并进入风险登记表。

示例输出

Attention模块:Q,K,V形状[B,L,d],einsum('bld,bmd->blm',Q,K)得到注意力分数,缩放因子需推断。

0

评论 0

更多

登录后可点赞、收藏、评论和举报。

还没有评论,先发起一个具体问题。