主要介绍了keras打印loss对权重的导数方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

Notes 怀疑模型梯度爆炸，想打印模型 loss 对各权重的导数看看。如果如果fit来训练的话，可以用keras.callbacks.TensorBoard实现。 但此次使用train_on_batch来训练的，用K.gradients和K.function实现。 Codes 以一份 VAE 代码为例 # -*- coding: utf8 -*- import keras from keras.models import Model from keras.layers import Inpu