Imported from https://github.com/hwalsuklee/tensorflow-mnist-MLP-batch_normalization-weight_initializers