LayerNormalization

功能

标准化张量

边界

【输入】

三个输入

X:数据类型为float16、float的张量

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

B:可选输入,数据类型为float16、float的张量,指定偏移量

【输出】

三个输出

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

mean:可选输出,均值

InvStdDev:可选输出,标准差的倒数

【属性】

axis:int,默认值为-1,第一个开始标准化的维度

epsilon:float,默认值为1e-05,指定一个小值与var相加,以避免除以0

stash_type:int,默认值为1,指定Mean和InvStdDev的类型