BatchNormalization

功能

标准化张量

边界

【输入】

五个输入

X:数据类型为float16、float32的4D张量

scale:数据类型为float32的张量,指定尺度因子

B:数据类型为float32的张量,指定偏移量

mean:数据类型为float32的张量,指定均值

var:数据类型为float32的张量,指定方差

【输出】

五个输出

Y:标准化之后的张量,数据类型为float16或float32

mean:均值

var:方差

saved_mean:在训练过程中使用已保存的平均值来加快梯度计算

saved_var:在训练过程中使用已保存的方差来加快梯度计算

【属性】

epsilon:可选,数据类型:float32,指定一个小值与var相加,以避免除以0,默认为0.0001

momentum:float32,该参数暂不支持

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13