Struct StreamableConv1d

pub struct StreamableConv1d { /* private fields */ }

Implementations

impl StreamableConv1d

pub fn new( in_c: usize, out_c: usize, k_size: usize, stride: usize, dilation: usize, groups: usize, bias: bool, causal: bool, norm: Option<Norm>, pad_mode: PadMode, vb: VarBuilder<'_>, ) -> Result<Self>

pub fn reset_batch_idx( &mut self, batch_idx: usize, _batch_size: usize, ) -> Result<()>

Trait Implementations

impl Clone for StreamableConv1d

fn clone(&self) -> StreamableConv1d

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StreamableConv1d

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

Formats the value using the given formatter.

impl Module for StreamableConv1d

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

impl StreamingModule for StreamableConv1d

fn reset_state(&mut self)

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