fix the div 0 error of pixel_shuffle (#49996)

paddle/phi/infermeta/unary.cc

@@ -2533,6 +2533,10 @@ void PixelShuffleInferMeta(const MetaTensor& x,
                         "Input should be a 4-D tensor of format [N, C, H, W] "
                         "or [N, H, W, C], but got %u.",
                         input_dims.size()));
+  PADDLE_ENFORCE_NE(
+      upscale_factor,
+      0,
+      phi::errors::InvalidArgument("upscale_factor should not be 0."));
 
   const bool channel_last = (data_format == "NHWC");
 

python/paddle/fluid/tests/unittests/test_pixel_shuffle.py

@@ -227,6 +227,13 @@ class TestPixelShuffleError(unittest.TestCase):
 
         self.assertRaises(TypeError, error_upscale_factor)
 
+        def error_0_upscale_factor():
+            with paddle.fluid.dygraph.guard():
+                x = paddle.uniform([1, 1, 1, 1], dtype='float64')
+                pixel_shuffle = F.pixel_shuffle(x, 0)
+
+        self.assertRaises(ValueError, error_0_upscale_factor)
+
         def error_data_format():
             with paddle.fluid.dygraph.guard():
                 x = np.random.random([2, 9, 4, 4]).astype("float64")