我正在尝试使用张量流给定的归一化层。在该function中,有一个字段指定我们使用的是beta值还是gamma值。


  中心:如果为True,则减去beta。如果为False,则忽略beta
  
  scale:如果为True,则乘以gamma。如果为False,则不使用gamma。如果下一层是线性的(例如nn.relu),则可以禁用此功能,因为缩放可以由下一层完成。


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

我需要知道如何输入betagamma ..的值吗?

最佳答案

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

看起来您应该能够像这样通过betagamma传递keyword arguments**kwargs

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

关于python - 如何在tf.contrib.layers.batch_norm中给出beta和gamma,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/38998238/

10-12 21:16