def output(self, train=True):
outputs_info = [self.s0]
(outputs, updates) = theano.scan(
fn=self.one_step,
sequences=self.X,
outputs_info=outputs_info,
go_backwards=self.go_backwards
)
return outputs