我如何获得BatchNorm的总体(测试)矩/方差值。张流中的层?

时间:2019-03-27 05:55:03

标签: tensorflow batch-normalization

我想找到一种方法来获取张量流的批标准化层中的总体(检验)均值/方差值。

我现在正在使用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))

但是我想知道提取变量的简单方法。

有人可以告诉我解决问题的方法吗?

非常感谢您。

0 个答案:

没有答案