Forward function. step_inputs, state0, step_outputs, and state1 should each be a `.NestedMap` of tensor values. Each tensor must be of shape [batch_size ...]. The structure of NestedMaps are determined by the implementation. state0 and state1 must have exactly the same structure and
(self, theta, prepared_inputs, step_inputs, padding, state0)
source not stored for this graph (policy: none)
no outgoing calls