docs: update api references

XuehaiPan · XuehaiPan · commit 813a6696548a · 2022-11-19T22:55:09.000+08:00
diff --git a/docs/source/api/api.rst b/docs/source/api/api.rst
@@ -202,16 +202,18 @@ Hook
 Gradient Transformation
 =======================
 
-.. currentmodule:: torchopt.clip
+.. currentmodule:: torchopt
 
 .. autosummary::
 
     clip_grad_norm
+    nan_to_num
 
 Transforms
 ~~~~~~~~~~
 
 .. autofunction:: clip_grad_norm
+.. autofunction:: nan_to_num
 
 Optimizer Schedules
 ===================
