Regarding the issue with the CrossMamba block

[ZHUXIUJINChris]

Thank you for the excellent work!

In this block, how does a single gated Z control both paths? I couldn't find it in the code. How exactly is it connected? Could you tell me the specific location in the code?

[wVelpro]

I have the same question.

[alexhe101]

In function: mamba_inner_fn_no_out_proj
https://github.com/alexhe101/Pan-Mamba/blob/main/Mamba-block/mamba_simple.py#L19

[ZHUXIUJINChris]

Thank you for your response. Is this gating operation originally present in VisionMamba and Mamba? Will using this function twice separately in "V3" form this connection?

I only saw this place using Z in the underlying code.