如何扩展SimpleRNN或创建一个自定义图层,该图层不仅可以使用先前的输出y[n-1]
,还可以使用先前的输出y[n-2]
和先前的输入x[n-1]
,{{1 }}?
x[n-2]
我的目标是以满足以下等式的方式重写输出:
def call(self, inputs, states):
prev_output = states[0]
h = K.dot(inputs, self.kernel)
output = h + K.dot(prev_output, self.recurrent_kernel)
return output, [output]