export layers

Author: liuanji

src/pyjuice/__init__.py

@@ -12,6 +12,9 @@
 # TensorCircuit
 from pyjuice.model import compile, TensorCircuit
 
+# TensorCircuit layers
+from pyjuice.layer import InputLayer, ProdLayer, SumLayer
+
 # Construction methods
 from pyjuice.nodes import multiply, summate, inputs, set_block_size, structural_properties