在tf.keras.layers.LSTM中,有一个名为activation
的参数(默认值为tanh
)。
然而,CuDNNLSTM没有这个参数,但却有一个名为activity_regularizer
的参数。
我是否遗漏了什么?
CuDNNLSTM不是tf.keras.layers.LSTM的包装器,而是一个独立的实体,那么我该如何为它设置激活函数呢?
在tf.keras.layers.LSTM中,有一个名为activation
的参数(默认值为tanh
)。
然而,CuDNNLSTM没有这个参数,但却有一个名为activity_regularizer
的参数。
我是否遗漏了什么?
CuDNNLSTM不是tf.keras.layers.LSTM的包装器,而是一个独立的实体,那么我该如何为它设置激活函数呢?
x = CuDNNLSTM(256, return_sequences=True)(x)
x = TimeDistributed(Dense(8))(x)
x = Softmax(axis=2)(x)