Перепишите tf.contrib.layers.batch_norm в Tensorflow 2.0

Question

Может ли кто-нибудь помочь мне переписать следующий блок кода в Tf2.0? Я знаю, что batch_norm эквивалентен keras.layers.BatchNormalization, но документация не дает четкого решения относительно того, чему соответствуют слова «decay» и «epsilon». Спасибо!

def batch_norm(opts, _input, is_train, reuse, scope, scale=True):
    """Batch normalization based on tf.contrib.layers.

    """
    return tf.contrib.layers.batch_norm(
        _input, center=True, scale=scale,
        epsilon=opts['batch_norm_eps'], decay=opts['batch_norm_decay'],
        is_training=is_train, reuse=True, updates_collections=None,
        scope=scope, fused=False)

Answer 1

В этом случае decay соответствует momentum из tf.keras.layers.BatchNormalization, а epsilon по-прежнему epsilon.