Theano softmax for tensors?

Thank you, That worked.