Possible to only inline preview of the first line for each variable

[zhuoqun-chen]

I have a use case where I need to know the shapes of each torch.Tensor, I have a custom hack for pytorch in python:

import torch
torch_available = True
def torch_custom_repr(self):
    return f">>> {tuple(self.shape)}\n{torch_original_repr(self)}\non <{self.device}>\n<<< {tuple(self.shape)}"

torch_original_repr = torch.Tensor.__repr__
torch.Tensor.__repr__ = torch_custom_repr

Now it looks like this in debug mode:

Is it possible to make it shorter like this:

[rchiodo]

Not sure what the question is. You've created your own custom __repr__, if you want it to output something different, change it again?

[zhuoqun-chen]

Hi @rchiodo , thx for your quick replay, do you mean that no matter how I implement __repr__, it will not affect the debug window like this right?

If that is the case, I think my problem is solved. Because I just don't want it to print the actual values of the array, it's kind messy, I only need to print the shape of the tensors in inline preview.

[rchiodo]

Changing __repr__ should change the first level of an object when viewed in the debugger, but yes it shouldn't affect the children.

Your original screen shot showed the first level of a tensor, you'd change your custom __repr__ if you wanted that to look different.

[zhuoqun-chen]

Thank you very much! My problem is solved.