#ptflops

该工具用于计算神经网络中的理论乘法加法运算量，以及参数数量和逐层计算成本。工具支持两个后端：pytorch和aten。aten后端覆盖更多模型架构，而pytorch后端更适合CNN分析。通过设置verbose参数，可以获取未纳入复杂度计算的操作信息，通过ignore_modules选项则可以忽略特定模块，适用于研究用途。适用于Pytorch版本2.0及以上。