Nesterov Accelerated Momentum Optimizer - Coding problem | OneNoughtOne