こんにちは、双方向の LSTM レイヤーがあります。
class BiDirLSTMInput(Layer):
def __init__(self):
self.bidir_lstm = Bidirectional(
LSTM(32, return_sequences=True,return_state=True)
)
def call(self, input):
o, h1,h2, c1,c2 = self.bidir_lstm(input)
return [h1,h2]
ご覧のとおり、LSTM から非表示の状態を消費しているだけです (セルの状態ではありません)。
それが理由ですか、次の警告が表示されますか:
WARNING:tensorflow:Gradients do not exist for variables for (backward layer):
- lstm_cell_2/kernel:0',
- lstm_cell_2/recurrent_kernel:0'
- lstm_cell_2/bias:0'
これを無視することは論理的に聞こえません。このエラーに対処するにはどうすればよいですか?