TensorFlow中的BatchNormalization

Question

tf.nn.batch_normalization和tf.contrib.layers.batch_norm之间有什么区别？我检查了文档，但不明白一个是否比另一个好。

Answer 1

tf.contrib是指Tensorflow上的高级机器学习API。因此tf.contrib.layers.batch_norm基本上充当tf.nn.batch_normal化的包装器，使其更易于使用，但不如后者灵活。因此，根据应用程序，您可以使用符合您需求的任何内容。例如，在tf.nn.batch_norm的情况下，您必须自己给出均值和方差张量的符号链接，但tf.contrib.layers.batch_norm将为您处理。