Mamba

class ssm.model.Mamba(n_layers, normalization=True, **kwargs)

Bases: Module

forward(x)

Forward pass through the Mamba model. It iterates over the Mamba blocks and applies them sequentially to the input tensor.

Parameters:

x (torch.Tensor) – Input tensor.

Returns:

Output tensor.

Return type:

torch.Tensor