diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
index 52a87cd2278..b7d6530561e 100644
--- a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
+++ b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
@@ -9,6 +9,14 @@ bigdl.nano.pytorch.Trainer
     :undoc-members:
     :exclude-members: accelerator_connector, checkpoint_connector, reload_dataloaders_every_n_epochs, limit_val_batches, logger, logger_connector, state
 
+bigdl.nano.pytorch.InferenceOptimizer
+---------------------------
+
+.. autoclass:: bigdl.nano.pytorch.InferenceOptimizer
+    :members:
+    :undoc-members:
+    :exclude-members: 
+
 bigdl.nano.pytorch.TorchNano
 ---------------------------