NVIDIA interview question

Implement batch norm 2D layer