Added compatibility for complex numbers (enables quantum neural netwo…

…rks), and allows functions called as tensor properties to be tracked (real, imag, T, mT, H).
johnmarktaylor91 · Feb 12, 2025 · 8ec9c89 · 8ec9c89
1 parent 42e93e1
commit 8ec9c89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name="torchlens",
-    version="0.1.30",
+    version="0.1.31",
     description="A package for extracting activations from PyTorch models",
     long_description="A package for extracting activations from PyTorch models. Contains functionality for "
                      "extracting model activations, visualizing a model's computational graph, and "

diff --git a/torchlens/helper_funcs.py b/torchlens/helper_funcs.py
@@ -404,7 +404,7 @@ def extend_search_stack_from_item(
 
     for attr_name in dir(item):
         if ((attr_name.startswith("__")) or
-                (attr_name in ['T', 'mT', 'real', 'imag']) or
+                (attr_name in ['T', 'mT', 'real', 'imag', 'H']) or
                 ('grad' in attr_name)):
             continue
         try: