From: François Fleuret Date: Mon, 5 Dec 2022 08:27:21 +0000 (+0100) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=c17cb30d6d5bfa2f450dd9cc3d2d931fce9cbdab;p=pytorch.git Update. --- diff --git a/stack.py b/stack.py index 0838bee..544306c 100755 --- a/stack.py +++ b/stack.py @@ -10,6 +10,11 @@ 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 its shape, + dtype and device. + + ''' repr_orig=Tensor.__repr__ Tensor.__repr__=lambda x: f'{x.size()}:{x.dtype}:{x.device}'