TensorFlowのバッチ正規化

Question

tf.nn.batch_normalizationとtf.contrib.layers.batch_normの違いは何ですか？私は文書をチェックしたが、他のものより優れているかどうかは分からなかった。

Answer 1

tf.contribは、Tensorflow上の高度なマシン学習APIを指します。したがって、tf.contrib.layers.batch_normは基本的にtf.nn.batch_normalization上のラッパーとして機能し、使用するのが簡単になりますが、後者より柔軟性が低くなります。したがって、用途に応じて、ニーズに合ったものを使用することができます。たとえば、tf.nn.batch_normの場合、あなた自身でmeanとvarianceテンソルにシンボリックリンクを与える必要がありますが、tf.contrib.layers.batch_normがそれを処理します。