diff --git a/python/paddle/autograd/functional.py b/python/paddle/autograd/functional.py
index 4d7fcd733cdb0b44e932de74f75b148ebc741d7b..17c7ad5b18af5f3e9b3b07f1f5c98fbf6ab56c49 100644
--- a/python/paddle/autograd/functional.py
+++ b/python/paddle/autograd/functional.py
@@ -16,7 +16,7 @@ import contextlib
 import paddle
 from ..fluid import framework
 from ..fluid.dygraph import grad
-from ..nn.initializer import assign
+from ..tensor.creation import assign
 from ..tensor import reshape, zeros_like, to_tensor
 from .utils import _tensors, _stack_tensor_or_return_none, _replace_none_with_zero_tensor