BatchNormalization in TensorFlow

Question

В чем разница между tf.nn.batch_normalization и tf.contrib.layers.batch_norm? Я проверил документацию, но не понял, лучше ли другой.

Answer 1

tf.contrib относится к высокоуровневому API для обучения машинам по Tensorflow. Таким образом, tf.contrib.layers.batch_norm в основном выступает в качестве обертки над tf.nn.batch_normalization, что упрощает его использование, но менее гибко, чем последнее. Таким образом, в зависимости от приложения вы можете использовать все, что соответствует вашим потребностям. Например, в случае tf.nn.batch_norm вам придется давать символическую ссылку на средние и тензоры дисперсии самостоятельно, но tf.contrib.layers.batch_norm позаботится об этом для вас.