ugh, stupid bug insoftmax.
gradient tester should really return a Tensor.