espnet2.asr.state_spaces.s4.SSKernelDiag

source

class espnet2.asr.state_spaces.s4.SSKernelDiag(A, B, C, log_dt, L=None, disc='bilinear', real_type='exp', lr=None, bandlimit=None)

Bases: OptimModule

Version using (complex) diagonal state matrix (S4D).

Initializes internal Module state, shared by both nn.Module and ScriptModule.

default_state(*batch_shape)

forward(L, state=None, rate=1.0, u=None)

Forward pass.

state: (B, H, N) initial state rate: sampling rate factor L: target length

returns: (C, H, L) convolution kernel (generally C=1) (B, H, L) output from initial state

forward_state(u, state)

step(u, state)