配置 PyCharm 调试器以显示数组/张量形状？

Question

当我使用 PyCharm 进行调试时，我希望调试器显示我的 NumPy 数组/Jax 数组/PyTorch 张量的形状。相反，我看到了他们的价值观：

有没有办法配置 PyCharm 的调试器，以便显示这些多维数组的形状？

Answer 1

我通过破解成员函数来实现这一点torch.Tensor.__repr__，因为 PyCharm 调试器调用__repr__以获取对象的字符串表示。

import torch

old_repr = torch.Tensor.__repr__
def tensor_info(tensor):
    return repr(tensor.shape)[6:] + ' ' + repr(tensor.dtype)[6:] + '@' + str(tensor.device) + '\n' + old_repr(tensor)
torch.Tensor.__repr__ = tensor_info

在 PyCharm 调试器中，您将看到如下表示：

>>>print(torch.ones(3,3))
Size([3, 3]) float32@cpu
tensor([[1., 1., 1.],
        [1., 1., 1.],
        [1., 1., 1.]])

我的变量检查器的快照：