|
|
|
@ -613,7 +613,6 @@ class BatchNorm(PrimitiveWithInfer):
|
|
|
|
|
- **updated_bias** (Tensor) - Tensor of shape :math:`(C,)`.
|
|
|
|
|
- **reserve_space_1** (Tensor) - Tensor of shape :math:`(C,)`.
|
|
|
|
|
- **reserve_space_2** (Tensor) - Tensor of shape :math:`(C,)`.
|
|
|
|
|
- **reserve_space_3** (Tensor) - Tensor of shape :math:`(C,)`.
|
|
|
|
|
|
|
|
|
|
Examples:
|
|
|
|
|
>>> input_x = Tensor(np.ones([128, 64, 32, 64]), mindspore.float32)
|
|
|
|
|