shgh:

elements of state_batch list are state. They are numbers (scalars)


So, state_batch is a list of 64 elements which are purely scalars? So, state_batch[0].shape would return 1? If that’s true, and you’re sure state_size is 3 then something’s wrong with the state_batch variable!