发明公开
- 专利标题: 一种针对softmax函数的高精度近似计算装置
-
申请号: CN202410295350.9申请日: 2024-03-15
-
公开(公告)号: CN118733946A公开(公告)日: 2024-10-01
- 发明人: 焦海龙 , 付宇哲
- 申请人: 北京大学深圳研究生院
- 申请人地址: 广东省深圳市南山区西丽镇丽水路深圳大学城北大校区
- 专利权人: 北京大学深圳研究生院
- 当前专利权人: 北京大学深圳研究生院
- 当前专利权人地址: 广东省深圳市南山区西丽镇丽水路深圳大学城北大校区
- 代理机构: 厦门市首创君合专利事务所有限公司
- 代理商 连耀忠
- 主分类号: G06F17/17
- IPC分类号: G06F17/17 ; G06F5/01 ; G06F7/498 ; G06F7/556 ; G06N3/063
摘要:
本发明公开了一种针对softmax函数的高精度近似计算装置,包括第一模块、第二模块和第三模块,第一模块包括第一零值预测单元、第一softmax函数计算单元和第一选择器单元,第二模块包括第二softmax函数计算单元,第三模块包括第二零值预测单元、第三softmax函数计算单元和第二选择器单元,第一零值预测单元和第二零值预测单元用于根据其对应输入的数据预测该装置的输出数据是否为零值,以进行跳除后续计算。第一softmax函数计算单元、第二softmax函数计算单元和第三softmax函数计算单元用于实现softmax函数的高精度计算,第一选择器单元和第二选择器单元用于选择最后的输出数据。本发明能够极大降低计算装置的功耗、面积等开销,提高装置能效。