Pycharm使用手册

2024/11/30 10:50:14 来源：https://blog.csdn.net/qq_42693593/article/details/142051685 浏览: 次关键词：Pycharm使用手册

Debugger不显示tensor的shape

默认的debugger不显示torch.Tensor的shape，导致debug效率较低。如下图所示：

在这里插入图片描述

可以在代码开头增加以下代码：

normal_repr = torch.Tensor.__repr__
torch.Tensor.__repr__ = lambda self: f"{self.shape}_{normal_repr(self)}"

之后显示为：

在这里插入图片描述

本网仅为发布的内容提供存储空间，不对发表、转载的内容提供任何形式的保证。凡本网注明“来源：XXX网络”的作品，均转载自其它媒体，著作权归作者所有，商业转载请联系作者获得授权，非商业转载请注明出处。

我们尊重并感谢每一位作者，均已注明文章来源和作者。如因作品内容、版权或其它问题，请及时与我们联系，联系邮箱：809451989@qq.com，投稿邮箱：809451989@qq.com