diff --git a/autograd/numpy/__init__.py b/autograd/numpy/__init__.py index 28b3f7a8..bacf7381 100644 --- a/autograd/numpy/__init__.py +++ b/autograd/numpy/__init__.py @@ -1,2 +1,15 @@ +from numpy import ( + diag_indices, + diag_indices_from, + fill_diagonal, + index_exp, + mgrid, + ndenumerate, + ndindex, + ogrid, + ravel_multi_index, + unravel_index, +) + from . import fft, linalg, numpy_boxes, numpy_jvps, numpy_vjps, numpy_vspaces, random from .numpy_wrapper import *