add tensorcircuit in api

Tractables · Feb 14, 2024 · 73c9c45 · 73c9c45
1 parent 2de7ce5
commit 73c9c45
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -23,3 +23,4 @@ API
 
    python-api/pyjuice
    python-api/nodes
+   python-api/tensorcircuit
diff --git a/src/pyjuice/model/tensorcircuit.py b/src/pyjuice/model/tensorcircuit.py
@@ -35,6 +35,10 @@ def _pc_model_backward_hook(grad, pc, inputs, record_cudagraph, apply_cudagraph,
 
 
 class TensorCircuit(nn.Module):
+    """
+    Test.
+    """
+
     def __init__(self, root_ns: CircuitNodes, layer_sparsity_tol: float = 0.5, 
                  max_num_partitions: Optional[int] = None, disable_gpu_compilation: bool = False, 
                  force_gpu_compilation: bool = False,
Original file line number	Diff line number	Diff line change
Expand Up		@@ -23,3 +23,4 @@ API

		python-api/pyjuice
		python-api/nodes
		python-api/tensorcircuit