add stg pages in sphinx (#1092)

aobo-y · facebook-github-bot · commit 7c228acbdf7e · 2022-12-16T12:39:29.000-08:00
Summary: Add sphinx pages for STG Pull Request resolved: #1092 Reviewed By: vivekmig Differential Revision: D42053425 Pulled By: aobo-y fbshipit-source-id: 93f098734ecc736c1b1232a8abb80d46c2362940
diff --git a/sphinx/source/binary_concrete_stg.rst b/sphinx/source/binary_concrete_stg.rst
@@ -0,0 +1,5 @@
+BinaryConcreteStochasticGates
+====================================
+
+.. autoclass:: captum.module.BinaryConcreteStochasticGates
+    :members:
diff --git a/sphinx/source/gaussian_stg.rst b/sphinx/source/gaussian_stg.rst
@@ -0,0 +1,5 @@
+GaussianStochasticGates
+====================================
+
+.. autoclass:: captum.module.GaussianStochasticGates
+    :members:
diff --git a/sphinx/source/index.rst b/sphinx/source/index.rst
@@ -19,6 +19,7 @@ Captum API Reference
    robust
    concept
    influence
+   module
    utilities
    base_classes
 
diff --git a/sphinx/source/module.rst b/sphinx/source/module.rst
@@ -0,0 +1,6 @@
+Module
+====================
+.. toctree::
+
+    binary_concrete_stg
+    gaussian_stg
