tf.contrib.layers.batch_norm可以像纸一样计算二维卷积批量标准化吗?

时间:2017-05-24 14:42:25

标签: tensorflow batch-normalization

如批量标准化的原始论文中所述,1-D特征(例如,来自完全连接的层)和2-D特征(例如,来自卷积层)的批量标准化在不平凡的方式。

tensorflow库提供了一种使用1-D功能批量标准化的简便方法,但我不确定它是否与2-D的情况相同。该工具为tf.contrib.layers.batch_norm

我不完全理解这种方法,但是我们可以将这种方法应用于二维批量标准化吗?

我看到有些人在二维特征地图上使用它(有多个频道):示例1(link 1link 2)。

Source page

1 个答案:

答案 0 :(得分:0)

您可以检查batch_normalization here的使用情况,或在Tensorflow 1.0之后搜索fused_bn的使用情况。