tensorflow - How to get weights in tf.layers.dense? -
i wanna draw weights of tf.layers.dense in tensorboard histogram, not show in parameter, how that?
the weights added variable named kernel
, use
x = tf.dense(...) weights = tf.get_default_graph().get_tensor_by_name( os.path.split(x.name)[0] + '/kernel:0'))
you can replace tf.get_default_graph()
other graph working in.
Comments
Post a Comment