From: François Fleuret <francois@fleuret.org>
Date: Mon, 28 Nov 2022 12:23:19 +0000 (+0100)
Subject: Update.
X-Git-Url: https://ant.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=1e0d48d059ab8c38f0d65cce169a57af2466d63b;p=pytorch.git

Update.
---

diff --git a/stack.py b/stack.py
index 27de204..453e9ed 100755
--- a/stack.py
+++ b/stack.py
@@ -11,13 +11,13 @@ import sys
 
 def exception_hook(exc_type, exc_value, tb):
 
-    tb = tb.tb_next
+#    tb = tb.tb_next
 
     while tb:
-        #x=tb.tb_frame.f_code
+        # x=tb.tb_frame #.f_code
         # for field in dir(x):
             # print(f'@@@ {field} {getattr(x, field)}')
-
+        print('--------------------------------------------------')
         filename = tb.tb_frame.f_code.co_filename
         name = tb.tb_frame.f_code.co_name
         line_no = tb.tb_lineno
@@ -28,7 +28,7 @@ def exception_hook(exc_type, exc_value, tb):
 
         for n,v in local_vars.items():
             if is_tensor(v):
-                print(f'  {n} -> {v.size()}:{v.dtype}:{v.device}')
+                print(f'  {n} -> {tuple(v.size())}:{v.dtype}:{v.device}')
             else:
                 print(f'  {n} -> {v}')
 
@@ -51,8 +51,9 @@ if __name__ == '__main__':
         c=b+b
         dummy(a,c)
 
-    m=torch.randn(2,3)
-    x=torch.randn(3)
-    blah(m,x)
-    blah(x,m)
+    mmm=torch.randn(2,3)
+    xxx=torch.randn(3)
+    #print(xxx@mmm)
+    blah(mmm,xxx)
+    blah(xxx,mmm)