关于z激活，在代码里面的体现在？

alexhe101 / Pan-Mamba

Pan-Mamba: Effective Pan-Sharpening with State Space Model

https://arxiv.org/abs/2402.12192

80 stars 7 forks source link

Closed zzduoYI closed 5 months ago

zzduoYI commented 5 months ago

a9d54e3318d5f7e3d02dc1bee57ecbe 这个我没看到在代码中的体现呀，能否告知一下？在此表示感谢

alexhe101 commented 5 months ago

这是mamba原始论文里的实现，我们直接调用了该函数，可以参考https://github.com/hustvl/Vim/blob/main/mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py#L606