Struct SeaNetResnetBlock

pub struct SeaNetResnetBlock { /* private fields */ }

Implementations

impl SeaNetResnetBlock

pub fn new( dim: usize, k_sizes_and_dilations: &[(usize, usize)], activation: Activation, norm: Option<Norm>, causal: bool, pad_mode: PadMode, compress: usize, true_skip: bool, vb: VarBuilder<'_>, ) -> Result<Self>

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

Trait Implementations

impl Clone for SeaNetResnetBlock

fn clone(&self) -> SeaNetResnetBlock

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SeaNetResnetBlock

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

Formats the value using the given formatter.

impl Module for SeaNetResnetBlock

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

impl StreamingModule for SeaNetResnetBlock

fn reset_state(&mut self)

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