BatchNormalization在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會照顧的，對於你。