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.
'''