如何在Tensorflow中更改LSTM单元格中的激活功能

Question

我正在尝试从Tensorflow的新1.0版本更改LSTM单元格中的激活功能，但是遇到了困难。

有一些API声明应该允许更改功能的tf.contrib.rnn.LSTMcell，但是这个单元格似乎还没有实现。

此外，tf.contrib.rnn.BasicLSTMCell，也应该允许不同的激活功能似乎不再存在。

我只需要等待还是有另一种解决方案？

Answer 1

当您实例化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)