Merge branch 'noklam-master' into master

parrt · Oct 10, 2020 · cec8fb6 · cec8fb6
2 parents f2e89ed + 2c46a54
commit cec8fb6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tsensor/analysis.py b/tsensor/analysis.py
@@ -24,7 +24,6 @@
 import os
 import sys
 import traceback
-import torch
 import inspect
 import hashlib
 
@@ -456,8 +455,8 @@ def istensor(x):
 
 def _shape(v):
     # do we have a shape and it answers len()? Should get stuff right.
-    if hasattr(v, "shape") and hasattr(v.shape,"__len__"):
-        if isinstance(v.shape, torch.Size):
+    if hasattr(v, "shape") and hasattr(v.shape, "__len__"):
+        if v.shape.__class__.__module__ == "torch" and v.shape.__class__.__name__ == "Size":
             if len(v.shape)==0:
                 return None
             return list(v.shape)