如何在TensorBoard中显示一个张量中多个要素的分布

Question

我有一个张量X，它是批量标准化层（tf.layers.batch_normalization）的输出，形状为[batch_size, 15]。为了监控其分布，我使用X创建了tf.summary.histogram('out_BN_0', X)的直方图。该图是我在Tensorboard中获得的＆gt; 70k步（~130个epoches）。这是所有15个功能的平均结果吗？或X中的任何特定功能？如何只分配一个功能（例如第5个）？

Answer 1

如何根据每个特征构建直方图？

import tensorflow as tf
import numpy as np

batch_size = 100
num_features = 15

X = tf.constant(np.random.uniform(size=(batch_size, num_features)))
hists = {feature_index: tf.summary.histogram(f'hist_{feature_index}', X[:, feature_index]) 
              for feature_index in range(num_features)}