Would it be better to import it from theano.compat instead?