diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 29517e1254a08bc518e5be31b07778b92968949a..e1a42710187823bb6ffd61849ef418757878e476 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -1270,6 +1270,13 @@ def batch_norm(
         eps: a value added to the denominator for numerical stability. Default: 1e-5
         inplace: whether to update ``running_mean`` and ``running_var``
             inplace or return new tensors. Default: True
+        compute_mode: When set to 'default', no special requirements will be
+            placed on the precision of intermediate results. When set to 'float32',
+            float32 would be used for accumulator and intermediate result, but only
+            effective when input and output are of float16 dtype.
+        param_dim: a value indicating in which format the parameters are.
+            Default: 'dim_1c11', which means NCHW format.
+            And 'dim_111c' means NHWC format.
     """
 
     def make_full_if_none(x, value):