BatchNormalization

功能

标准化tensor。

输入

五个输入：

X：数据类型为float16、float的4Dtensor。

scale：可选，数据类型为float的tensor，指定尺度因子。

B：可选，数据类型为float的tensor，指定偏移量。

mean：可选，数据类型为float的tensor，指定均值。

var：可选，数据类型为float的tensor，指定方差。

输出

五个输出：

Y：标准化之后的tensor，数据类型为float16或float。

mean：可选，数据类型为float的tensor，均值。

var：可选，数据类型为float的tensor，方差。

saved_mean：可选，数据类型为float的tensor，在训练过程中使用已保存的平均值来加快梯度计算（v14后官方不支持）。

saved_var：可选，数据类型为float的tensor，在训练过程中使用已保存的方差来加快梯度计算（v14后官方不支持）。

属性

epsilon：可选，数据类型：float，指定一个小值与var相加，以避免除以0，默认为0.0001。

momentum：float，该参数暂不支持。

支持的ONNX版本

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