python - 如何在 tf.contrib.layers.batch_norm 中给出 beta 和 gamma

Question

我正在尝试使用 tensorflow 给出的归一化层。在那function ，有一个字段指定我们是使用 beta 还是 gamma 值。

center: If True, subtract beta. If False, beta is ignored.

scale: If True, multiply by gamma. If False, gamma is not used. When the next layer is linear (also e.g. nn.relu), this can be disabled since the scaling can be done by the next layer.

但是没有地方可以在函数中输入 beta 和 gamma 值。

我需要知道如何输入 beta 和 gamma 的值..？

Answer 1

您提供的链接已将函数指定为tf.contrib.layers.batch_norm(*args, **kwargs)。

看起来您应该能够将 beta 和 gamma 作为 关键字参数 或 **kwargs 传递这个:

tf.contrib.layers.batch_norm(beta=value, gamma=value)