Browse Source

!14259 Fix BN docs

From: @jojobugfree
Reviewed-by: @zhoufeng54,@jjfeing
Signed-off-by: @jjfeing
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
2457fe476f
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ops/operations/nn_ops.py

+ 2
- 0
mindspore/ops/operations/nn_ops.py View File

@@ -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`.


Loading…
Cancel
Save