有效

一种实现Softmax函数的设备

赵云、曾勇刚、崔超、肖勇、林伟斌、徐迪
南方电网科学研究院有限责任公司

摘要

本发明提供了一种实现Softmax函数的设备,接口模块获取用于实现Softmax函数的多个输入数据;转换器模块对每个输入数据进行逻辑运算,得到每个输入数据对应的第一参数和第二参数并将其存储至存储模块;指数运算模块对存储模块中的第一参数和第二参数进行指数运算和移位运算,得到每个输入数据的以自然常数为底的指数函数运算值并将其发送至存储模块和累加模块;累加模块累加所有指数函数运算值得到累加值;除法模块从存储模块中逐个获取指数函数运算值,计算各个指数函数运算值和累加值的比值,得到每个输入数据对应的概率值。通过硬件的方式实现Softmax函数,不需要占用CPU的计算资源,提高神经网络模型的运算速度。