Revisit Policy Optimization in Matrix Form

Re-assign community