Pycharm使用手册

Debugger不显示tensor的shape

默认的debugger不显示torch.Tensor的shape，导致debug效率较低。如下图所示：

可以在代码开头增加以下代码：

normal_repr = torch.Tensor.__repr__
torch.Tensor.__repr__ = lambda self: f"{self.shape}_{normal_repr(self)}"

之后显示为：