BatchNormalization

功能

对输入Tensor进行批量归一化操作。

输入

• X：输入Tensor，数据类型支持float16、float的4Dtensor，数据格式支持ND。
• scale（可选）：输入Tensor，数据类型支持float，指定尺度因子，数据格式支持ND。
• B（可选）：输入Tensor，数据类型支持float，指定偏移量，数据格式支持ND。
• input_mean（可选）：输入Tensor，数据类型支持float，指定均值，数据格式支持ND。
• input_var（可选）：输入Tensor，数据类型支持float，指定方差，数据格式支持ND。

属性

• epsilon（可选）：数据类型：float，指定一个小值与var相加，以避免除以0，默认为0.0001。
• momentum：float，该参数暂不支持。

输出

• Y：标准化之后的tensor，数据类型为float16或float，数据格式支持ND。
• running_mean（可选）：数据类型为float的tensor，均值，数据格式支持ND。
• running_var（可选）：数据类型为float的tensor，方差，数据格式支持ND。
• saved_mean（可选）：数据类型为float的tensor，在训练过程中使用已保存的平均值来加快梯度计算（v14后官方不支持），数据格式支持ND。
• saved_var（可选）：数据类型为float的tensor，在训练过程中使用已保存的方差来加快梯度计算（v14后官方不支持），数据格式支持ND。

约束与限制

无。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18