Reshare input before equalize (#6775)

e2fa1f9d · Vasilis Vryniotis · GitHub · e1aacdd9 · e2fa1f9d
Unverified Commit e2fa1f9d authored Oct 14, 2022 by Vasilis Vryniotis Committed by GitHub Oct 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/prototype/transforms/functional/_color.py torchvision/prototype/transforms/functional/_color.py +1 -1

No files found.
--- a/torchvision/prototype/transforms/functional/_color.py
+++ b/torchvision/prototype/transforms/functional/_color.py
@@ -227,7 +227,7 @@ def equalize_image_tensor(image: torch.Tensor) -> torch.Tensor:
    if image.numel() == 0:
        return image

-    return _equalize_image_tensor_vec(image.view(-1, height, width)).reshape(image.shape)
+    return _equalize_image_tensor_vec(image.reshape(-1, height, width)).reshape(image.shape)


 equalize_image_pil = _FP.equalize