https://timodenk.com/blog/tensorflow-batch-normalization/