Struct FastGRNN 

pub struct FastGRNN { /* private fields */ }

FastGRNN model for neural routing

Implementations

impl FastGRNN

pub fn new(config: FastGRNNConfig) -> Result<Self>

Create a new FastGRNN model with the given configuration

pub fn load<P: AsRef<Path>>(_path: P) -> Result<Self>

Load model from a file (safetensors format)

pub fn save<P: AsRef<Path>>(&self, _path: P) -> Result<()>

Save model to a file (safetensors format)

pub fn forward( &self, input: &[f32], initial_hidden: Option<&[f32]>, ) -> Result<f32>

Forward pass through the FastGRNN model

Arguments

• input - Input vector (sequence of features)
• initial_hidden - Optional initial hidden state

Returns

Output score (typically between 0.0 and 1.0 after sigmoid)

pub fn forward_batch(&self, inputs: &[Vec<f32>]) -> Result<Vec<f32>>

Batch inference for multiple inputs

pub fn quantize(&mut self) -> Result<()>

Quantize the model to INT8

pub fn prune(&mut self, sparsity: f32) -> Result<()>

Apply magnitude-based pruning

pub fn size_bytes(&self) -> usize

Get model size in bytes

pub fn config(&self) -> &FastGRNNConfig

Get configuration