diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 639deba157d13c34aa8c08586185364220f34c94..bd25825af68bb56c213dece48122a8a78d7713a1 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -5852,8 +5852,8 @@ def sampled_softmax_with_cross_entropy(logits,
         inputs={
             'Logits': logits,
             'Labels': label,
-            'CustomSamples': custom_samples,
-            'CustomProbabilities': custom_probabilities
+            'CustomizedSamples': customized_samples,
+            'CustomizedProbabilities': customized_probabilities
         },
         outputs={
             'Samples': samples,