optax.contrib.SplitRealAndImaginaryState#
- class optax.contrib.SplitRealAndImaginaryState(inner_state: optax.OptState)[source]#
Maintains the inner transformation state for split_real_and_imaginary.
Maintains the inner transformation state for split_real_and_imaginary.