From 813a6696548aa0c7acca5c6e62e6592aac0f4d65 Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Sat, 19 Nov 2022 22:55:09 +0800
Subject: [PATCH] docs: update api references

---
 docs/source/api/api.rst | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docs/source/api/api.rst b/docs/source/api/api.rst
index 63950e30e..f2f4a6716 100644
--- 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
 ===================