Update NeuralNetwork.py

Appeared (by grad checking) that the additional layers were not being computed correctly. A corrected and more thoroughly checked version is posted here
Merge pull request #1 from ajcortese/updateEinsum

Update NeuralNetwork.py