Neural Network Trainable Parameter Counter - Coding problem | OneNoughtOne