MoE Top-K 路由前向

大模型手撕 · 困难 · MoE、模型结构

实现一个 token 级 Top-K 路由的 MoE 层(逐 token 计算即可):

数据范围

在 RagCodes 上用 Python / C++ / Java 在线编写并提交,核心代码模式与 ACM 模式均可评测。