如何计算lstm网络的复杂度 乘法次数 flops（未完成）

复杂度 乘法次数 flops（未完成）"/>

如何计算lstm网络的复杂度 乘法次数 flops（未完成）

计算算法复杂度的一种方法是计算前向算法的乘法次数，因此在这篇文章中计算复杂度的方法是计算lstm网络的乘法运算次数

首先要弄清楚lstm的cell中的乘法次数

cell有4个输入，一个输出

有三个门，input gate控制数据输不输进来，不输入就输入0

forget gate控制保存单元memory cell更不更新，不更新就维持原状

output gate控制计算的值输不输出，不输出就输出0

举例

x是input y是output

lstm的cell有4个input，就是矢量x+bias，乘上weight，求和

以信号处理为例，batch为1，全部省略，此时的x大小[1,Nv] 

接下来是用别人写好的库计算，用的是pytorch OpCounter

GitHub - Lyken17/pytorch-OpCounter: Count the MACs / FLOPs of your PyTorch model.

按照github上的提示

pip install thop

然后就可以运行下面的例子了