CNIPA.AI
검색으로 돌아가기
기록

基于嵌入式平台的softmax计算的优化方法、系统、介质及设备

발명유효
10청구항 · 4 독립항
§ Ⅰ

개요

발명자

汪双蕾; 匡启和; 郑茳; 肖佐楠; 王廷平

IPC 분류

G06F 17/10 (2006.01)G06N 3/048 (2023.01)

本申请公开了一种基于嵌入式平台的softmax计算的优化方法、系统、存储介质及设备,属于softmax算子优化技术领域,该方法包括:在嵌入式平台下,对输入数据序列进行搜索,确定输入数据序列包含的所有元素当中的最大值;计算输入数据序列中每一元素与最大值的差值,将所有的差值组成第一数据序列,并进行保存;判断第一数据序列中的每一差值与特定阈值的大小关系,并根据大小关系,得到第二数据序列;创建与输入数据序列的元素个数相同的位置掩码表;根据第二数据序列进行softmax计算,输出第一结果序列,并进行保存;根据第一结果序列和位置掩码表,生成第二结果序列。本申请在嵌入式平台下对softmax计算进行优化,提高了硬件资源利用率,兼顾了模型泛化性能和精度。