打印张量流模型中的激活次数



我正在尝试计算模型中的激活次数,例如在LeNet中。如何计算激活总数?

有一种方法可以计算可训练参数的数量,但是,似乎没有用于计算单个激活的选项。

激活次数取决于模型的层,例如:

  • 对于全连接层(Dense),激活次数等于神经元数量。
  • 对于卷积层,激活次数是过滤器的数量乘以输出特征图的空间维度(这取决于填充、输入大小等)。
  • 对于循环层,这取决于LSTM/GRU具有复杂的结构。对于一个简单的RNN,它只是神经元的数量乘以时间步的数量。

最新更新