在STM32上实现Keras中的LSTM网络

linxuplus

于 2021-11-08 19:50:48 发布

阅读量2k

点赞数 3

文章标签： stm32 keras lstm

本文链接：https://blog.csdn.net/ZXCVBNNHU/article/details/121214495

版权

本文介绍如何在STM32微控制器上实现Keras中的LSTM网络。通过在keras环境中训练参数，不使用LSTM的bias，然后导出权重到STM32。在STM32上，利用arm_math库的矩阵乘法运算，根据LSTM门的顺序进行计算，成功复现了Keras中的LSTM网络运算结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先在keras环境中训练参数，得到lstm的权重和偏置（此处推荐不选择使用LSTM的bias选项，这样可以在剪枝等操作时更好地对比变化后的影响）

①查看keras的LSTM计算过程

self.kernel_i = self.kernel[:, :self.units]
        self.kernel_f = self.kernel[:, self.units: self.units * 2]
        self.kernel_c = self.kernel[:, self.units * 2: self.units * 3]
        self.kernel_o = self.kernel[:, self.units * 3:]

        self.recurrent_kernel_i = self.recurrent_kernel[:, :self.units]
        self.recurrent_kernel_f = self.recurrent_kernel[:, self.units: self.units * 2]
        self.recurrent_kernel_c = self.recurrent_kernel[:, self.units * 2: self.units * 3]
        self.recurrent_kernel_o = self.recurrent_kernel[:, self.units * 3:]

        if self.use_bias:
            self.bias_i = self.bias[:self.units]
            self.bias_f = self.bias[self.units: self.units * 2]
            self.bias_c = self.bias[self.units * 2: self.units * 3]
            self.bias_o = self.b

最低0.47元/天解锁文章