tensorflow - 为什么 tensorflow 的 FLOPs 是 Caffe 的 macc 的 2 倍？

Question

我正在尝试将模型从 caffe 重写为 tensorflow。为了确保我没有犯错，我数了一下 macc 和 Flops，然后我发现了这个有趣的事情：

例如，当输入一个 112x112x3 的图像，并用 32 个 3x3 的 kernel，stride=1 对其进行 conv2d 处理时，Caffe 中的 macc 为 2.71M，而 tensorflow 中计算的 FLOPs 为 5. 42M。

我想知道为什么会发生这种 2 倍的差异？

标签： tensorflowcaffeflops

Answer 1

一个 MACC 包含一个乘法和一个加法。一次乘法或一次加法可以看作是一次 FLOP。所以一个 MACC 有两个 FLOPS。