mindchemistry.e3.nn.BatchNorm =============================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/r2.5.0/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindscience/blob/r0.7/docs/api_python/mindchemistry/e3/nn/mindchemistry.e3.nn.BatchNorm.rst :alt: 查看源文件 .. py:class:: mindchemistry.e3.nn.BatchNorm(irreps, eps=1e-5, momentum=0.1, affine=True, reduce='mean', instance=False, normalization='component', dtype=float32) 正交表示的批归一化。 通过表示的范数进行归一化。 注意,范数仅对正交表示是不变的。 不可约表示 `wigner_D` 是正交的。 参数: - **irreps** (Union[str, Irrep, Irreps]) - 输入的不可约表示。 - **eps** (float) - 归一化方差时避免除以零的值。默认值:``1e-5``。 - **momentum** (float) - 滑动平均的动量。默认值:``0.1``。 - **affine** (bool) - 是否包含权重和偏置参数。默认值:``True``。 - **reduce** (str) - {'mean', 'max'},用于归约的方法。默认值:``'mean'``。 - **instance** (bool) - 应用实例归一化而不是批归一化。默认值:``False``。 - **normalization** (str) - {'component', 'norm'},归一化方法。默认值:``'component'``。 - **dtype** (mindspore.dtype) - 输入张量的类型。默认值:``mindspore.float32`` 。 输入: - **input** (Tensor) - 形状为 :math:`(batch, ..., irreps.dim)` 的张量。 输出: - **output** (Tensor) - 形状为 :math:`(batch, ..., irreps.dim)` 的张量。 异常: - **ValueError**: 如果 `reduce` 不是 'mean' 或 'max'。 - **ValueError**: 如果 `normalization` 不是 'component' 或 'norm'。