Oh sorry, I did not understand at first.
Yeah, I also think this would be a good idea to print the shape of the tensor.
Here is the link to the file, there you can make the suggestion directly to add the shape to the _repr_.
Or maybe someone will see this issue here.
If not, you can tell me and I can try to make the suggestion myself 