Browse Source

fixed BatchNorm

tags/v0.5.0-beta
jiangjinsheng 5 years ago
parent
commit
7aa5e7f3c3
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      mindspore/ops/operations/nn_ops.py

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

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


Loading…
Cancel
Save