在运行的入口 py 文件里加入以下代码：

1
2
3
4
5
6
# show tensor shape in vscode debugger
def custom_repr(self):
    return f'{{Tensor:{tuple(self.shape)}}} {original_repr(self)}'

original_repr = torch.Tensor.__repr__
torch.Tensor.__repr__ = custom_repr

就会在调试中显示出 __repr__ 的替换内容，如图所示：
![[Pasted image 20240629103104.png|400]]