NVIDIA Interview Question

Implement batch norm 2D layer