Glorot/Xavier Weight Initialization - Coding problem | OneNoughtOne