variables.py 文件源码

def summarize_gradients(model_name, gradients):
    """ Adds histograms for gradients and gradient norms of the input
    gradients """

    def get_prefix(var):
        return model_name + '/' + var.name

    for gradient, variable in gradients:
        if gradient is not None:
            tf.summary.histogram(get_prefix(variable) + "/gradients", gradient)
            tf.summary.histogram(get_prefix(variable) + "/gradient_norm",
                                 tf.global_norm([gradient]))