模型训练时保存在tensorflow中的激活值在哪里?



在tensorflow2中,我们知道在反向传播过程中,我们需要权重和激活值来计算偏导数,我可以在模型tf.keras.Model中找到存储的权重,但是激活值(用于反向传播计算的中间值),它们存储在哪里?

tf.keras.activations下的激活检查API文档

最新更新