Normalize a tensor along the specific axis/dimension. The code is in main.py.
Requirements are Python >= 3.5 and TensorFlow 1.5.0. Requirements are specified in the pip's requirements.txt file.
Create virtualenv with Python 3.5. Install requirements with the command pip install -r requirements.txt and run code with python main.py.