Browse Source

!6870 refine FusedBatchNormEx doc format

Merge pull request !6870 from zyli2020/bug_fix
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
a95ed7e8e0
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ops/operations/nn_ops.py

+ 4
- 4
mindspore/ops/operations/nn_ops.py View File

@@ -654,11 +654,11 @@ class FusedBatchNormEx(PrimitiveWithInfer):

Inputs:
- **input_x** (Tensor) - The input of FusedBatchNormEx, Tensor of shape :math:`(N, C)`,
data type: float16 or float32.
data type: float16 or float32.
- **scale** (Tensor) - Parameter scale, same with gamma above-mentioned, Tensor of shape :math:`(C,)`,
data type: float32.
data type: float32.
- **bias** (Tensor) - Parameter bias, same with beta above-mentioned, Tensor of shape :math:`(C,)`,
data type: float32.
data type: float32.
- **mean** (Tensor) - mean value, Tensor of shape :math:`(C,)`, data type: float32.
- **variance** (Tensor) - variance value, Tensor of shape :math:`(C,)`, data type: float32.

@@ -670,7 +670,7 @@ class FusedBatchNormEx(PrimitiveWithInfer):
- **updated_bias** (Tensor) - Updated parameter bias, Tensor of shape :math:`(C,)`, data type: float32.
- **updated_moving_mean** (Tensor) - Updated mean value, Tensor of shape :math:`(C,)`, data type: float32.
- **updated_moving_variance** (Tensor) - Updated variance value, Tensor of shape :math:`(C,)`,
data type: float32.
data type: float32.
- **reserve** (Tensor) - reserve space, Tensor of shape :math:`(C,)`, data type: float32.

Examples:


Loading…
Cancel
Save