From: François Fleuret <francois@fleuret.org>
Date: Fri, 9 Dec 2022 05:51:33 +0000 (+0100)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=ca897077ed89fbc3c7e8d812ad262146a0c72b71

Update.
---

diff --git a/tensorstack.py b/tensorstack.py
index d7b0d89..c9a6c2f 100755
--- a/tensorstack.py
+++ b/tensorstack.py
@@ -10,9 +10,9 @@ from torch import Tensor
 import sys
 
 def exception_hook(exc_type, exc_value, tb):
-    r'''Hacks the call stack message in case of RuntimeError to show all
-    the local variables, and indicate for every tensor involved its
-    shape, dtype and device.
+    r'''Hacks the call stack message to show all the local variables in
+    case of RuntimeError or ValueError, and prints tensors as shape,
+    dtype and device.
 
     '''