simplify PIL comparisions

Author: justincdavis

test/test_transforms_v2.py

@@ -6480,11 +6480,7 @@ def test_image_correctness(self, num_output_channels, color_space, make_input, f
         actual = fn(image, num_output_channels=num_output_channels)
 
         if make_input is make_image_cvcuda:
-            actual = F.cvcuda_to_tensor(actual).to(device="cpu")
-            actual = actual.squeeze(0)
-            # drop the batch dimension
-            image = F.cvcuda_to_tensor(image).to(device="cpu")
-            image = image.squeeze(0)
+            image = cvcuda_to_pil_compatible_tensor(image)
 
         expected = F.to_image(F.rgb_to_grayscale(F.to_pil_image(image), num_output_channels=num_output_channels))
 
@@ -6585,11 +6581,7 @@ def test_image_correctness(self, make_input, fn):
         actual = fn(image)
 
         if make_input is make_image_cvcuda:
-            actual = F.cvcuda_to_tensor(actual).to(device="cpu")
-            actual = actual.squeeze(0)
-            # drop the batch dimension
-            image = F.cvcuda_to_tensor(image).to(device="cpu")
-            image = image.squeeze(0)
+            image = cvcuda_to_pil_compatible_tensor(image)
 
         expected = F.to_image(F.grayscale_to_rgb(F.to_pil_image(image)))