vscode调试时显示张量的维度shape

在程序的入口添加 如下代码即可：

import torch
def patch_tensor_repr():original_repr = torch.Tensor.__repr__def new_repr(self):shape_str = f"Shape: {tuple(self.shape)}"tensor_str = original_repr(self)return f"{shape_str} {tensor_str}"torch.Tensor.__repr__ = new_repr
patch_tensor_repr()