Tensorflow loss nan. However, I keep getting a loss: nan output.