我想找到一种方法来获取张量流的批标准化层中的总体(检验)均值/方差值。
我现在正在使用BN代码:
tf.nn.batch_normalization()
现在,我正在通过这些代码手动计算人口矩:
train_mean4 = tf.assign(pop_mean4, pop_mean4 * decay + batch_mean4 * (1 - decay))
train_var4 = tf.assign(pop_var4, pop_var4 * decay + batch_var4 * (1 - decay))
但是我想知道提取变量的简单方法。
有人可以告诉我解决问题的方法吗?
非常感谢您。