diff --git a/python/paddle/fluid/layers/loss.py b/python/paddle/fluid/layers/loss.py
index ed64c96eb6f33fc852c7916862212de2523920a9..eaac99fc5b5923b77a2b078f1a61f62c4efc395b 100644
--- a/python/paddle/fluid/layers/loss.py
+++ b/python/paddle/fluid/layers/loss.py
@@ -16,6 +16,7 @@ from __future__ import print_function
 
 import numpy as np
 from functools import partial, reduce
+import paddle
 from paddle.utils import deprecated
 from . import nn
 from .layer_function_generator import templatedoc