memorax.utils.Transition

class memorax.utils.Transition

Bases: object

Transition(first: memorax.utils.timestep.Timestep | None = None, second: memorax.utils.timestep.Timestep | None = None, carry: typing.Any | None = None, aux: typing.Any | None = None)

first: Timestep | None = None

second: Timestep | None = None

carry: Any | None = None

aux: Any | None = None

to_sequence()

Return type:

Self

from_sequence()

Return type:

Self

__init__(first=None, second=None, carry=None, aux=None)

Parameters:

• first (Timestep | None)

• second (Timestep | None)

• carry (Any | None)

• aux (Any | None)

Return type:

None

replace(**updates)

Returns a new object replacing the specified fields with new values.