[Doc] Add missing classes to doc

ghstack-source-id: 2e174577aa33cc8d69c0f423c90ea2e5ee0fdef6 Pull Request resolved: #1203
pytorch · Feb 3, 2025 · 5848f1f · 5848f1f
1 parent 1f35ab4
commit 5848f1f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/docs/source/reference/nn.rst b/docs/source/reference/nn.rst
@@ -194,6 +194,7 @@ to build distributions from network outputs and get summary statistics or sample
     TensorDictModuleBase
     TensorDictModule
     ProbabilisticTensorDictModule
+    ProbabilisticTensorDictSequential
     TensorDictSequential
     TensorDictModuleWrapper
     CudaGraphModule
@@ -257,6 +258,10 @@ Distributions
     NormalParamExtractor
     OneHotCategorical
     TruncatedNormal
+    InteractionType
+    set_interaction_type
+    add_custom_mapping
+    mappings
 
 
 Utils
@@ -270,8 +275,8 @@ Utils
 
     make_tensordict
     dispatch
-    set_interaction_type
     inv_softplus
     biased_softplus
     set_skip_existing
     skip_existing
+    rand_one_hot