ARTICLE DETAIL

资讯详情

深耕网站建设、视觉设计与SEO优化的一线实战洞察。

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

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

  

 

返回列表