Trait StreamingModule

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

Streaming modules take as input a stream tensor and return a stream tensor. They may perform some internal buffering so that enough data has been received for the module to be able to perform some operations.

Required Methods

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

fn reset_state(&mut self)

Implementors

impl<T: Module> StreamingModule for Map<T>