[improvement] using directly numpy arrays instead of theano