Added scripted fn save test for random erase (pytorch#2767)

Author: vfdev-5

test/test_functional_tensor.py

@@ -741,10 +741,6 @@ def test_perspective(self):
                         batch_tensors, F.perspective, startpoints=spoints, endpoints=epoints, interpolation=0
                     )
 
-    def test_convert_image_dtype(self):
-        # TODO: add tests of CPU/CUDA on tensor and batch
-        pass
-
 
 @unittest.skipIf(not torch.cuda.is_available(), reason="Skip if no CUDA device")
 class CUDATester(Tester):

test/test_transforms_tensor.py

@@ -490,6 +490,9 @@ def test_random_erasing(self):
             self._test_transform_vs_scripted(fn, scripted_fn, tensor)
             self._test_transform_vs_scripted_on_batch(fn, scripted_fn, batch_tensors)
 
+        with get_tmp_dir() as tmp_dir:
+            scripted_fn.save(os.path.join(tmp_dir, "t_random_erasing.pt"))
+
     def test_convert_image_dtype(self):
         tensor, _ = self._create_data(26, 34, device=self.device)
         batch_tensors = torch.rand(4, 3, 44, 56, device=self.device)