Как визуализировать на Tensorboard гистограммы ввода и вывода функции активации с информацией о тензорном потоке и градиенте keras?

Question

Я хотел бы добавить несколько гистограмм в тензорную таблицу. В частности, я хотел бы визуализировать для каждого слоя гистограмму ввода и вывода его функции активации.

В настоящее время я визуализирую веса и смещения, используя keras.callbacks.TensorBoard:

tensorboard_cb = keras.callbacks.TensorBoard(logdir, histogram_freq=1, write_graph=True, write_grads=True, write_images=True)

и повторение каждого уровня модели:

for i,l in enumerate(m.layers[1:]):
    w, b = l.get_weights()
    tf.summary.histogram(name='W%d'%i, data=w, step=1, description="weigths of " +l.get_config()['name'])
    tf.summary.histogram(name='B%d'%i, data=b, step=1, description="biases of " +l.get_config()['name'])

, где m - обученная модель.

которая работает. Однако, несмотря на настройку with_grads=True, информация о градиенте не отображается на Tensorboard