diff --git a/python/mxnet/optimizer.py b/python/mxnet/optimizer.py index f758af5f982c..7e69cf6131f6 100644 --- a/python/mxnet/optimizer.py +++ b/python/mxnet/optimizer.py @@ -24,7 +24,7 @@ import warnings import numpy from .base import py_str -from .ndarray import (NDArray, zeros, clip, sqrt, cast, maximum, abs as NDabs) +from .ndarray import (NDArray, zeros, clip, sqrt, cast, maximum, abs as NDabs, array, multiply) from .ndarray import (sgd_update, sgd_mom_update, adam_update, rmsprop_update, rmspropalex_update, mp_sgd_update, mp_sgd_mom_update, square, ftrl_update, ftml_update, signsgd_update, signum_update)