pub type GruForwardOutput<F> = (Array<F, IxDyn>, (Array<F, IxDyn>, Array<F, IxDyn>, Array<F, IxDyn>));
Type alias for GRU forward output (new_h, (reset_gate, update_gate, new_gate))