Tensorboard激活直方图

Question

如何在张量板中绘制tf.layers.dense激活的直方图？请考虑以下情况：

with tf.variable_scope('hidden'):
hidden_layer = tf.layers.dense(input, 100, activation=tf.nn.relu)

我可以使用以下代码绘制hidden的权重和偏差：

for v in tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES, scope='hidden'):
    tf.summary.histogram("{}/weight_histogram".format(v.name), v)

但是，我似乎也无法获得激活的集合。我怎么能修改它来绘制激活的直方图？

Answer 1

在您的模型范围内，您需要调用以下内容：tf.contrib.layers.summarize_activation(hidden_layer)