- 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.
+ r"""Hacks the call stack message to show all the local variables
+ in case of RuntimeError, ValueError, or TypeError and prints
+ tensors as shape, dtype and device.