之前咨询过百度的大咖，回复是截至1.7版本没有专门的函数
但是我发现了可以用下面的方法

import paddleslim as slim
FLOPs = slim.analysis.flops(train_program)
print("FLOPs: {}".format(FLOPs))

更多推荐

paddle paddle如何统计FLOPs