From 882d848272875ee70fe1503d9dc74e6aea967f32 Mon Sep 17 00:00:00 2001 From: caifubi Date: Sat, 27 Mar 2021 18:22:03 +0800 Subject: [PATCH] fix BN docs --- mindspore/ops/operations/nn_ops.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index fd388725a9..6b87d68efd 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -1129,6 +1129,7 @@ class BatchNorm(PrimitiveWithInfer): Inputs: If `is_training` is False, inputs are Tensors. + - **input_x** (Tensor) - Tensor of shape :math:`(N, C)`, with float16 or float32 data type. - **scale** (Tensor) - Tensor of shape :math:`(C,)`, with float16 or float32 data type. - **bias** (Tensor) - Tensor of shape :math:`(C,)`, has the same data type with `scale`. @@ -1136,6 +1137,7 @@ class BatchNorm(PrimitiveWithInfer): - **variance** (Tensor) - Tensor of shape :math:`(C,)`, has the same data type with `mean`. If `is_training` is True, `scale`, `bias`, `mean` and `variance` are Parameters. + - **input_x** (Tensor) - Tensor of shape :math:`(N, C)`, with float16 or float32 data type. - **scale** (Parameter) - Parameter of shape :math:`(C,)`, with float16 or float32 data type. - **bias** (Parameter) - Parameter of shape :math:`(C,)`, has the same data type with `scale`.