diff --git a/python/paddle/nn/layer/loss.py b/python/paddle/nn/layer/loss.py
index ef2e22ce60624..8ede3440ae1ef 100644
--- a/python/paddle/nn/layer/loss.py
+++ b/python/paddle/nn/layer/loss.py
@@ -1340,10 +1340,16 @@ class CTCLoss(Layer):
     blank: int
     reduction: _ReduceMode
 
-    def __init__(self, blank: int = 0, reduction: _ReduceMode = 'mean') -> None:
+    def __init__(
+        self,
+        blank: int = 0,
+        reduction: _ReduceMode = 'mean',
+        zero_infinity: bool = False,
+    ) -> None:
         super().__init__()
         self.blank = blank
         self.reduction = reduction
+        self.zero_infinity = zero_infinity
 
     def forward(
         self,
@@ -1352,7 +1358,6 @@ def forward(
         input_lengths: Tensor,
         label_lengths: Tensor,
         norm_by_times: bool = False,
-        zero_infinity: bool = False,
     ) -> Tensor:
         return paddle.nn.functional.ctc_loss(
             log_probs,
@@ -1362,7 +1367,7 @@ def forward(
             self.blank,
             self.reduction,
             norm_by_times=norm_by_times,
-            zero_infinity=zero_infinity,
+            zero_infinity=self.zero_infinity,
         )