mlx_rs::nn

Type Alias GruInput

Source
pub type GruInput<'a> = RnnInput<'a>;
Expand description

Type alias for the input of the GRU module.

Aliased Type§

struct GruInput<'a> {
    pub x: &'a Array,
    pub hidden: Option<&'a Array>,
}

Fields§

§x: &'a Array

Input tensor

§hidden: Option<&'a Array>

Hidden state