diff --git a/docs/mindspore/_ext/overwrite_autodoc.txt b/docs/mindspore/_ext/overwrite_autodoc.txt
index acbad745367b0a711c5bfe9d3f3c4c06bf7c9bd4..ba9ae3881609fa5f2a4a193b55439c08e54246d8 100644
--- a/docs/mindspore/_ext/overwrite_autodoc.txt
+++ b/docs/mindspore/_ext/overwrite_autodoc.txt
@@ -1801,7 +1801,10 @@ class ClassDocumenter(DocstringSignatureMixin, ModuleLevelDocumenter):  # type:
                 mint_doc = re.sub(r'(\n[ ]+(-|\*) GPU(/CPU)?:.*\n|\n[ ]+(-|\*) CPU(/GPU)?:.*\n)', '\n', docstrings[i])
                 mint_doc = re.sub(r'(Alias for.*?:(func|class):`[^`]+?\.)ops\.([^`]+?)(?<!_ext)(`|_ext`)', r'\1mint.\3`', mint_doc)
                 mint_doc = re.sub(r'(Supported Platforms:\n\s+?)`.*\n', r'\1``Ascend``\n', mint_doc)
-                mint_doc = re.sub('(>>> from mindspore import .*?)(ops|nn)', r'\1mint', mint_doc)
+                if 'mindspore/mint/optim' in py_source_rel:
+                    mint_doc = re.sub('(>>> from mindspore import .*?)(ops)', r'\1mint', mint_doc)
+                else:
+                    mint_doc = re.sub('(>>> from mindspore import .*?)(ops|nn)', r'\1mint', mint_doc)
                 mint_doc = mint_doc.replace('>>> import mindspore.ops as ops', '>>> from mindspore import mint')
                 if not re.findall(f'>>> .*?{usename}', mint_doc):
                     mint_doc = re.sub(rf'(>>> .*?)(ops|nn)\.[\w\.]*?{b_name}(Ext)?', rf'\1{usename}', mint_doc)