diff --git a/mmedit/models/common/linear_module.py b/mmedit/models/common/linear_module.py
index d3417a88e1..5101ad1643 100644
--- a/mmedit/models/common/linear_module.py
+++ b/mmedit/models/common/linear_module.py
@@ -73,7 +73,7 @@ def forward(self, x, activate=True):
         """Forward Function.
 
         Args:
-            x (torch.Tensor): Input tensor with shape of (n, *, c).
+            x (torch.Tensor): Input tensor with shape of :math:`(n, *, c)`.
                 Same as ``torch.nn.Linear``.
             activate (bool, optional): Whether to use activation layer.
                 Defaults to True.