Policy gradient

shape