https://yehoon17.github.io/posts/batch_norm/