VSCODE调试显示变量维度

在代码开头写入:

import torch
def custom_repr(self):
    return f'{{Tensor:{tuple(self.shape)}}} {original_repr(self)}'

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

你可能感兴趣的:(python,深度学习,vscode)