Function repeat_frame

rl2/a3c/worker.py:41–42

(frame)

# Create initial state by repeating the same frame 4 times
def repeat_frame(frame):
  return np.stack([frame] * 4, axis=2)


# Create next state by shifting each frame by 1