Struct StreamingTransformer 

pub struct StreamingTransformer { /* private fields */ }

Implementations

impl StreamingTransformer

pub fn new(cfg: &Config, vb: VarBuilder<'_>) -> Result<Self>

pub fn forward(&mut self, xs: &Tensor) -> Result<Tensor>

pub fn forward_ca( &mut self, xs: &Tensor, ca_src: Option<&Tensor>, ) -> Result<Tensor>

pub fn copy_state(&mut self, from: &Self) -> Result<()>

Trait Implementations

impl Clone for StreamingTransformer

fn clone(&self) -> StreamingTransformer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for StreamingTransformer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl StreamingModule for StreamingTransformer

fn reset_state(&mut self)

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