当前位置: 首页 > news >正文

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

http://www.hskmm.com/?act=detail&tid=36007

相关文章:

  • 软工第三次作业--结对作业
  • Day1HTML的基本骨架
  • 树上依赖性背包 学习笔记 | P6326 Shopping 题解
  • java标识符
  • 题解:uoj961【UR #30】赛场设计
  • 位运算快速卷积 快速沃尔什变换 FWT
  • 嵌合抗体:破解二抗选择难题,赋能多重分子检测的核心工具
  • 原来用聊天记录就可以创造数字分身!WeClone项目在Lab4AI平台上的复现
  • 自监督提示优化SPO
  • Java中的注释
  • 实测!不同场景下,哪款 AI IDE 能真正帮你少加班?
  • CSP-S模拟36 2025.10.21
  • 2025 年 AI 编程工具生成效果全景比拼:从技术实力到综合评分
  • 打造AI IDE标杆产品,腾讯CodeBuddy深度全方位解析
  • C语言项目开发常用目录结构 - Invinc
  • 2025年不锈钢水箱厂家权威推荐榜:方形/圆形/消防/生活/保温/承压/装配式/焊接水箱及水塔水罐全解析
  • day03-Coze记忆-对话体验
  • 2025年流量计厂家权威推荐榜单:电磁流量计、超声波流量计、涡街流量计、质量流量计专业制造商深度解析
  • RNDIS让Air8000的USB上网更智能、更快速!
  • 如果k8s有三个calico节点A,B,C 使用bgp模式的话是如何进行BGP对等会话的
  • 10.21
  • home-assistant-Onboarding Home Assistant(入职家庭助理)
  • Day1标签语法
  • home-assistant-Concepts and terminology概念和术语
  • 2025年印染水洗机厂家权威推荐榜:高效水洗设备与环保节能技术深度解析,专业水洗机厂家精选
  • 高级语言程序设计第二次作业
  • 有关K8s calico IPIP模式的一些疑惑和思考
  • 1.正手握拍
  • 2025年角接触轴承厂家推荐排行榜,高精度/高承载/高精密/机床主轴/汽车/定制/可替代进口/高转速/高刚性角接触球轴承公司推荐
  • 7-Zip最新版 7-Zip25.01