金融界2024年12月21日消息,国家知识产权局信息显示,北京君正集成电路股份有限公司申请一项名为“一种基于SIMD指令的LSTM实现方法”的专利,公开号CN119150938A,申请日期为2023年6月。
专利摘要显示,本发明提供一种基于SIMD指令的LSTM实现方法,包括:S1第一层循环是时间序列的循环,每次对一个时间序列的值进行计算得出结果,一共循环time_sequence次;S2第二层循环是输出通道的循环,输出通道有output_channel个,一共是output_channel/16次循环;每次循环先计算4种权重的数据指针;每次循环需要把sum_I,sum_J,sum_F,sum_O重新置0;S3第三层循环分成两个循环,第一个循环是对input_data数据循环。第二个循环是对input_hidden数据循环;S4,第三层循环结束后在第二层循环中对sum_I,sum_J,sum_F,sum_O继续进行计算。第二层循环结束,已经得到当前时间序列的所有输出outputchannel的信息再进行下次时间序列的循环即第一层循环,直至第一层循环结束。该方法通过将SIMD指令组合使用完成与Python实现的LSTM相同的功能。
本文源自:金融界