VSCode直观显示tensor形状

''' A simple example to show tensor shape on debugger ''' 
from _pydevd_bundle.pydevd_extension_api import StrPresentationProvider 
from .pydevd_helpers import find_mod_attr 

class PyTorchTensorShapeStr: 
	def can_provide(self, type_object, type_name): 
		torch_tensor = find_mod_attr('torch', 'Tensor') 
		return torch_tensor is not None and issubclass(type_object, torch_tensor) 
		
	def get_str(self, val): 
		dim = val.dim() 
		if dim == 0: 
			return str(val) 
		elif dim == 1: 
			return f'{val.shape[0]}: {val}' 
		return f'{list(val.shape)}: {val}' 

import sys 

if not sys.platform.startswith("java"): 
	StrPresentationProvider.register(PyTorchTensorShapeStr)

放到下，随后重启 VSCode 即可解决
~/.vscode/extensions/ms-python.debugpy-XXX/bundled/libs/debugpy/_vendored/pydevd/pydevd_plugins/extensions/types/pydevd_plugin_pytorch_tensor_str.py
。

由于 debugpy 版本不同，路径中的 XXX 部分会不一样；如果是 VSCode Remote 环境，放到远程机器的 ~/.vscode-server/... 下的相同子目录即可。

wsl2 下是 ~/.vscode-server，如果用过了 debug 模式会自动生成 extensions文件夹，且安装好了debugpy，如"~/.vscode-server/extensions/ms-python.debugpy-2024.14.0-linux-x64"

参考自： https://www.zhihu.com/question/560178647