Apple interview question

Implement batch norm using Numpy