- class returnn.frontend.state.State(*args, **kwargs)#
Covers all the state of a recurrent module, i.e. exactly what needs to be stored and passed into the module next time you call it as initial state.
This behaves somewhat like a namedtuple, although we derive from dict.
When you derive further from this class, make sure that it works correctly with
tree, which creates new instances of the same class by calling
keys_and_values = ((key, result[key]) for key in instance). See
LstmStatefor an example:
class LstmState(rf.State): def __init__(self, *_args, h: Tensor = None, c: Tensor = None): super().__init__(*_args) if not _args: self.h = h self.c = c