|
|
|
@@ -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: |
|
|
|
|