LayerNormalization

功能

标准化tensor。

输入

三个输入：

X：数据类型为float16的tensor。

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

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

输出

三个输出：

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

mean：可选输出，均值。

InvStdDev：可选输出，标准差的倒数。

属性

axis：int，默认值为-1，第一个开始标准化的维度。

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

stash_type：int，默认值为1，指定Mean和InvStdDev的类型。

支持的ONNX版本

Opset v17