我正在尝试从新的 Tensorflow 1.0 版本中更改 LSTM 单元中的激活函数,但遇到了困难。
有 tf.contrib.rnn.LSTMcell,API 状态应该允许更改功能,但似乎尚未为此单元实现。
此外,tf.contrib.rnn.BasicLSTMCell,它也应该允许不同的激活函数似乎不再存在。
我只需要等待还是有其他解决方案?
当您实例化两者时tf.contrib.rnn.LSTMcell
,tf.contrib.rnn.BasicLSTMCell
您可以将激活函数作为activation
参数传递。如果您查看链接的文档,您会看到,例如,构造函数的签名BasicLSTMCell
是
__init__(num_units, forget_bias=1.0, input_size=None, state_is_tuple=True, activation=tf.tanh)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句