1mindspore.mint.nn.functional.batch_norm 2======================================= 3 4.. py:function:: mindspore.mint.nn.functional.batch_norm(input, running_mean, running_var, weight=None, bias=None, training=False, momentum=0.1, eps=1e-5) 5 6 对输入数据进行批量归一化和更新参数。 7 8 批量归一化广泛应用于卷积神经网络中。此运算对输入应用归一化，避免内部协变量偏移，详见论文 `Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift <https://arxiv.org/abs/1502.03167>`_ 。使用mini-batch数据和学习参数进行训练，学习的参数见如下公式中， 9 10 .. math:: 11 y = \frac{x - mean}{\sqrt{variance + \epsilon}} * \gamma + \beta 12 13 其中， :math:`\gamma` 为 `weight`， :math:`\beta` 为 `bias`， :math:`\epsilon` 为 `eps`， :math:`mean` 为 :math:`x` 的均值， :math:`variance` 为 :math:`x` 的方差。 14 15 参数： 16 - **input** (Tensor) - 数据输入，shape为 :math:`(N, C, *)` 的Tensor，其中 :math:`*` 表示任意的附加维度。对于Atlas 训练系列产品，当前仅支持2-4维。支持数据类型为bfloat16，float16或float32。 17 - **running_mean** (Tensor) - shape为 :math:`(C,)` ，数据类型为bfloat16，float16或float32。 18 - **running_var** (Tensor) - shape为 :math:`(C,)` ，数据类型为bfloat16，float16或float32。 19 - **weight** (Tensor, 可选) - shape为 :math:`(C,)` ，数据类型为bfloat16，float16或float32，默认值: ``None`` 。为 ``None`` 时，初始化为 ``1`` 。 20 - **bias** (Tensor, 可选) - shape为 :math:`(C,)` ，数据类型为bfloat16，float16或float32，默认值: ``None`` 。为 ``None`` 时，初始化为 ``0`` 。 21 - **training** (bool，可选) - 如果 `training` 为 `True`，`running_mean` 和 `running_var` 会在训练过程中进行计算。 22 如果 `training` 为 `False` ，它们会在推理阶段从checkpoint中加载。默认值： ``False`` 。 23 - **momentum** (float, 可选) - 动态均值和动态方差所使用的动量。（例如 :math:`new\_running\_mean = (1 - momentum) * running\_mean + momentum * current\_mean`）。默认值： ``0.1`` 。 24 - **eps** (float, 可选) - 添加到分母上的值，以确保数值稳定性。默认值： ``1e-5``。 25 26 返回： 27 Tensor，数据类型与shape大小与 `input` 相同，其中，shape大小为 :math:`(N, C, *)` 。 28 29 异常： 30 - **TypeError** - `training` 不是bool。 31 - **TypeError** - `eps` 或 `momentum` 的数据类型不是float。 32 - **TypeError** - `input`、`weight`、`bias`、`running_mean` 或 `running_var` 不是Tensor。