text: # For reward condition right 
  A[3][:,:,1] = [ 0 .5 
  0 .5  0 .5 
  0 .5  0 .5 
  0 .5  1 .0 
  0 .0 ] 
  
  # For reward condition left 
  A[3][:,:,2] = [ 0 .5 
  0 .5  0 .5 
  0 .5  0 .5 
  0 .5  0 .0 
  1 .0 ] 
  (cid:6)  (cid:5) 
  
  Having created all three modalities of**A**, we could continue to**B**, or the transition model. 
  Each of the two state factors of**B**—the location factor and reward condition factor—needed 
  to be defined separately. We started with the location factor, which contained the transition 
  to and from four possible location states under four different actions: a 4 × 4 × 4 tensor. 
  The agent could control these states perfectly with its four movement actions, independently 