Trait StreamingModule

Source
pub trait StreamingModule {
    // Required methods
    fn step(
        &mut self,
        xs: &StreamTensor,
        mask: &StreamMask,
    ) -> Result<StreamTensor>;
    fn reset_state(&mut self);
}

Required Methods§

Source

fn step(&mut self, xs: &StreamTensor, mask: &StreamMask) -> Result<StreamTensor>

Source

fn reset_state(&mut self)

Implementors§