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()