The original code shows as following: import theano.tensor as tt
import theano.tensor as tt r = inp[:, 0] # response/choice in current trial for all subjects o
