Changed produceRandomGradOutput to handle number outputs correctly
Replaced torch.randn(1)[1] by torch.normal() call
Merge pull request #5 from akfidjeland/number-output

Changed produceRandomGradOutput to handle number outputs correctly