Fixes memory reset for GRUs

Author: Mayankm96

rsl_rl/modules/actor_critic_recurrent.py

@@ -98,4 +98,4 @@ def reset(self, dones=None):
         if self.hidden_states is None:
             return
         for hidden_state in self.hidden_states:
-            hidden_state[..., dones == 1] = 0.0
+            hidden_state[..., dones == 1, :] = 0.0