如何在 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)