Tensor Shape-Shifting: See Shapes Before Values

Awesome hack:

Want to see the shapes of tensor first while debugging in VSCode?

Just run this:

torch.Tensor.repr = lambda self: f"{tuple(self.shape)}:{normal_repr(self)}"

It replaces Tensor’s repr so that it shows shape first!

Discussion