Fix warning in import

2021-09-28 17:11:31 +08:00 · 2021-09-28 17:11:31 +08:00 · 526711accd
parent 8cb632a6ec
commit 526711accd
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/init.py
+++ b/mindspore/nn/init.py
@ -26,7 +26,7 @@ from .loss import *
 from .optim import *
 from .metrics import *
 from .wrap import *
-from .grad import *
+from .grad import Jvp, Vjp
 from .sparse import *