lower a little bit theano overhead by removing call to enumerate as is it slow. Anyway, the index was already available otherwise.