如何在 Tensorflow 中更改 LSTM 单元中的激活函数
How to change the activation function in LSTM cell in Tensorflow
我正在尝试从新的 1.0 版 Tensorflow 更改 LSTM 单元中的激活函数,但遇到困难。
tf.contrib.rnn.LSTMcell API 状态应该允许更改功能,但它似乎尚未针对此单元实现。
此外,tf.contrib.rnn.BasicLSTMCell,它也应该允许不同的激活函数似乎不再存在。
我是只需要等待还是有其他解决方案?
当您实例化两个 tf.contrib.rnn.LSTMcell
and tf.contrib.rnn.BasicLSTMCell
时,您可以将激活函数作为 activation
参数传递。如果您查看链接的文档,您会看到 BasicLSTMCell
的构造函数签名是
__init__(num_units, forget_bias=1.0, input_size=None, state_is_tuple=True, activation=tf.tanh)
我正在尝试从新的 1.0 版 Tensorflow 更改 LSTM 单元中的激活函数,但遇到困难。
tf.contrib.rnn.LSTMcell API 状态应该允许更改功能,但它似乎尚未针对此单元实现。
此外,tf.contrib.rnn.BasicLSTMCell,它也应该允许不同的激活函数似乎不再存在。
我是只需要等待还是有其他解决方案?
当您实例化两个 tf.contrib.rnn.LSTMcell
and tf.contrib.rnn.BasicLSTMCell
时,您可以将激活函数作为 activation
参数传递。如果您查看链接的文档,您会看到 BasicLSTMCell
的构造函数签名是
__init__(num_units, forget_bias=1.0, input_size=None, state_is_tuple=True, activation=tf.tanh)