|
|
|
@@ -23,15 +23,15 @@ mindspore.nn.BatchNorm2d |
|
|
|
|
|
|
|
**参数:** |
|
|
|
|
|
|
|
- **num_features** (int) - 指定输入Tensor的通道数量,输入Tensor的size为(N, C, H, W)。 |
|
|
|
- **eps** (float) - 加在分母上的值,以确保数值稳定。默认值:1e-5。 |
|
|
|
- **num_features** (int) - 指定输入Tensor的通道数量,输入Tensor的size为 :math:`(N, C, H, W)` 。 |
|
|
|
- **eps** (float) - :math:`\epsilon` 加在分母上的值,以确保数值稳定。默认值:1e-5。 |
|
|
|
- **momentum** (float) - 动态均值和动态方差所使用的动量。默认值:0.9。 |
|
|
|
- **affine** (bool) - bool类型。设置为True时,可学习gama和beta值。默认值:True。 |
|
|
|
- **gamma_init** (Union[Tensor, str, Initializer, numbers.Number]) - gamma参数的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'ones'。 |
|
|
|
- **beta_init** (Union[Tensor, str, Initializer, numbers.Number]) - beta参数的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'zeros'。 |
|
|
|
- **affine** (bool) - bool类型。设置为True时,可学习 :math:`\gamma` 和 :math:`\beta` 值。默认值:True。 |
|
|
|
- **gamma_init** (Union[Tensor, str, Initializer, numbers.Number]) - :math:`\gamma` 参数的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'ones'。 |
|
|
|
- **beta_init** (Union[Tensor, str, Initializer, numbers.Number]) - :math:`\beta` 参数的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'zeros'。 |
|
|
|
- **moving_mean_init** (Union[Tensor, str, Initializer, numbers.Number]) - 动态平均值的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'zeros'。 |
|
|
|
- **moving_var_init** (Union[Tensor, str, Initializer, numbers.Number]) - 动态方差的初始化方法。str的值引用自函数 `mindspore.common.initializer`,包括'zeros'、'ones'等。默认值:'ones'。 |
|
|
|
- **use_batch_statistics** (bool) - 如果为True,则使用当前批处理数据的平均值和方差值,并跟踪运行平均值和运行方差。如果为False,则使用指定值的平均值和方差值,不跟踪统计值。如果为None,则根据训练和验证模式自动设置use_batch_statistics为True或False。在训练时,use_batch_statistics会设置为True。在验证时,use_batch_statistics会自动设置为False。默认值:None。 |
|
|
|
- **use_batch_statistics** (bool) - 如果为True,则使用当前批处理数据的平均值和方差值,并跟踪运行平均值和运行方差。如果为False,则使用指定值的平均值和方差值,不跟踪统计值。如果为None,则根据训练和验证模式自动设置 `use_batch_statistics` 为True或False。在训练时, `use_batch_statistics会` 设置为True。在验证时, `use_batch_statistics` 会自动设置为False。默认值:None。 |
|
|
|
- **data_format** (str) - 数据格式可为'NHWC'或'NCHW'。默认值:'NCHW'。 |
|
|
|
|
|
|
|
**输入:** |
|
|
|
|