[0-size Tensor No.166、257] Add 0-size Tensor support for paddle.nn.functional.ctc_loss [fluid_ops] (PaddlePaddle#74042)

* Fix

* Fix

* Fix

* ci

* Fix

* Fix

* Fix

Author: co63oc

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/multiary_infer_sym.cc

@@ -4489,6 +4489,17 @@ bool WarpctcOpInferSymbolicShape(
       infer_context->GetShapeOrDataForValue(op->operand_source(0));
   const std::vector<symbol::DimExpr> &logits_shape =
       logits_shape_or_data.shape();
+  bool logits_0_size = false;
+  for (size_t i = 0; i < logits_shape.size(); ++i) {
+    if (logits_shape[i] == 0) {
+      logits_0_size = true;
+      break;
+    }
+  }
+  if (logits_0_size) {
+    PADDLE_THROW(
+        common::errors::InvalidArgument("The input size can not be zero."));
+  }
 
   symbol::DimExpr max_sequence_length, num_sequences;
   symbol::DimExpr sequence_width = symbol::DimExpr(1);

paddle/phi/infermeta/multiary.cc

@@ -5645,6 +5645,9 @@ void WarpctcInferMeta(const MetaTensor& logits,
                       MetaTensor* loss,
                       MetaTensor* warpctcgrad) {
   auto logits_dims = logits.dims();
+  if (common::product(logits_dims) == 0) {
+    PADDLE_THROW(errors::InvalidArgument("The input size can not be zero."));
+  }
   int num_sequences, sequence_width, max_sequence_length;
 
   if (logits_length && labels_length) {

python/paddle/signal.py

@@ -569,6 +569,7 @@ def istft(
     fft_size = x.shape[-2]
 
     if in_dynamic_mode():
+        assert x.size != 0, 'x should not be an empty tensor.'
         if onesided:
             assert (
                 fft_size == n_fft // 2 + 1

test/legacy_test/test_signal.py

@@ -1045,5 +1045,12 @@ def test_istft(self):
             ),
 
 
+class TestIstftException_ZeroSize(unittest.TestCase):
+    def test_istft(self):
+        self.x = np.random.random([5, 0])
+        with self.assertRaises(AssertionError):
+            paddle.signal.istft(paddle.to_tensor(self.x), 512)
+
+
 if __name__ == '__main__':
     unittest.main()

test/legacy_test/test_warpctc_op.py

@@ -613,8 +613,24 @@ def test_dygraph_with_lod():
                 reduction='none',
             )
 
+        def test_dygraph_zero_size():
+            logits = np.random.uniform(0.1, 1.0, [0, 15]).astype("float32")
+            # labels should not be blank
+            labels = np.random.randint(0, 15 - 1, [15, 1], dtype="int32")
+            softmax = paddle.to_tensor(logits)
+            labels = paddle.to_tensor(labels)
+
+            paddle.nn.functional.ctc_loss(
+                log_probs=softmax,
+                labels=labels,
+                input_lengths=None,
+                label_lengths=None,
+                reduction='none',
+            )
+
         paddle.disable_static()
         self.assertRaises(ValueError, test_dygraph_with_lod)
+        self.assertRaises(ValueError, test_dygraph_zero_size)
         paddle.enable_static()