Skip to content

GitLab

Unverified Commit 7f55738e authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

remove custom __repr__ from prototype features (#5916)

parent 734ee253
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
torchvision/prototype/features/_feature.py
View file @ 7f55738e
......@@ -83,6 +83,3 @@ class _Feature(torch.Tensor):
return cls.new_like(args[0], output, dtype=output.dtype, device=output.device)
else:
return output
def __repr__(self) -> str:
return cast(str, torch.Tensor.__repr__(self)).replace("tensor", type(self).__name__)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment