about A = -torch.exp(self.A_log.float())

johnma2006 / mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

Apache License 2.0

2.54k stars 188 forks source link

Open liang00fan opened 3 months ago

liang00fan commented 3 months ago

why add "-" in this code?why not use A directly instead of A_log? A = -torch.exp(self.A_log.float())

liang00fan commented 3 months ago