How to get weights in tf.layers.dense?

Question

I wanna draw the weights of tf.layers.dense in tensorboard histogram, but it not show in the parameter, how could I do that?

Answer 1

In TF 2 if you're inside a @tf.function (graph mode):

weights = optimizer.weights

If you're in eager mode (default in TF2 except in @tf.function decorated functions):

weights = optimizer.get_weights()