买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:深圳市九天睿芯科技有限公司
摘要:本发明涉及一种用于大语言模型的softmax计算方法、装置及存储介质。所述方法包括:获取大语言模型输出的全部token,并对全部token的分值进行排序,根据排序结果选取分值位于预设区间的token并将剩余的token剔除,以得到处理后的分类向量,将分类向量从浮点型数值量化为第一比特宽度的整数,并将第一比特宽度的整数输入至softmax分类器中进行softmax运算,得到softmax概率值,将运算得到的softmax概率值量化为第二比特宽度的整数,以得到分类结果。本发明中的softmax算法引入了量化技术,大幅度降低了计算量,提升了大语言模型的推理效率。在特定的softmax的硬件实现架构下,提高了llama模型在实际应用中的性能和效率。
主权项:1.一种用于大语言模型的softmax计算方法,其特征在于,所述方法包括:获取大语言模型输出的全部token,并对所述全部token的分值进行排序;根据排序结果选取分值位于预设区间的token并将剩余的token剔除,以得到处理后的分类向量;将所述分类向量从浮点型数值量化为第一比特宽度的整数,并将所述第一比特宽度的整数输入至softmax分类器中进行softmax运算,以得到softmax概率值;将运算得到的softmax概率值量化为第二比特宽度的整数,以得到分类结果。
全文数据:
权利要求:
百度查询: 深圳市九天睿芯科技有限公司 用于大语言模型的softmax计算方法、装置及存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。