Update torchrl/modules/tensordict_module/rnn.py

Co-authored-by: Vincent Moens <vincentmoens@gmail.com>
pytorch · Dec 1, 2023 · 9e84a04 · 9e84a04
1 parent 1ccfafa
commit 9e84a04
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torchrl/modules/tensordict_module/rnn.py b/torchrl/modules/tensordict_module/rnn.py
@@ -149,7 +149,10 @@ def lstm_cell(self, x, hx, cx):
 
 
 class LSTM(nn.LSTM):
-    """A module that runs multiple steps of a multi-layer LSTM and is only coded in Python."""
+    """A module that runs multiple steps of a multi-layer LSTM and is only coded in Python.
+
+    """
+    __doc__ += nn.LSTM.__doc__
 
     def __init__(
         self,