xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.config#
Classes#
Module Contents#
- class xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.config.mLSTMBackend(config)#
Bases:
torch.nn.Module- Parameters:
config (Any)
- config#
- abstract forward(q, k, v, i, f)#
- Parameters:
q (torch.Tensor)
k (torch.Tensor)
v (torch.Tensor)
i (torch.Tensor)
f (torch.Tensor)
- Return type: