2025年3月3日
DeepSeek MOE 代码实现
摘要： 前置知识： PyTorch 基础函数操作整理 1. topk 操作 功能: torch.topk 用于返回输入张量中指定维度上的前 k 个最大元素及其对应的索引。 示例代码: import torch x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5] 阅读全文
posted @ 2025-03-03 18:57 AAA建材王师傅 阅读(394) 评论(3) 推荐(0)