Как я могу получить значение популяции (тест) моментов / дисперсии BatchNorm.слой в тензорном потоке? - PullRequest
0 голосов
/ 27 марта 2019

Я хочу найти способ получить значение совокупности (тест) среднего значения / дисперсии в слое батчнормализации тензорного потока.

Сейчас я использую код 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))

Но я хочу знать простой способ извлечения переменных.

Может кто-нибудь сказать мне путьрешить проблему?

Большое вам спасибо.

...