首先贴上官方文档
γ和β初始化时默认为1和0.输入必须是4维的张量x,而对四维张量进行归一化的步骤是: 对不同样本的同一个通道的所有像素进行求均值. 最后得到的均值是一个(1,a,1,1) 的张量, a为通道数量. 即对于每一个通道,都能计算出一个均值. 然后再求方差
, 最后进行变换
得到输出.
首先执行一段代码:
然后推导计算过程:
计算结果与程序输出一致
首先贴上官方文档
γ和β初始化时默认为1和0.输入必须是4维的张量x,而对四维张量进行归一化的步骤是: 对不同样本的同一个通道的所有像素进行求均值. 最后得到的均值是一个(1,a,1,1) 的张量, a为通道数量. 即对于每一个通道,都能计算出一个均值. 然后再求方差
, 最后进行变换
得到输出.
首先执行一段代码:
然后推导计算过程:
计算结果与程序输出一致