About data flipping

[zzzendurance]

Thank you for reading my question. In the code, the data is flipped in the last dimension: xz.flip([-1]), so I want to ask if it can be changed in other dimensions (you only need to answer me this question), because I am worried that if I only change xz.flip([-1]), the code logic elsewhere is related to this code logic.

[zzzendurance]

I have just tried to flip another dimension and found that it works successfully, but I am not sure if this change is effective. If the author sees my question, thank you for your answer

[Karn3003]

Hi @zzzendurance , It should be 1 right? As I do think in out_b it should be out_b.flip([1]) instead of -1. As we need to flip the tokens not the dimensions across the tokens.

[zzzendurance]

Hi @zzzendurance , It should be 1 right? As I do think in out_b it should be out_b.flip([1]) instead of -1. As we need to flip the tokens not the dimensions across the tokens.

I think what you said makes sense. I’ve been following this code for a long time, and I almost forgot about it. I just looked it up, and it should be this line of code. This line is likely flipping the c dimension in the (b, t, c) format, right?

Vim/mamba-1p1p1/mamba_ssm/modules/mamba_simple.py

Line 230 in 1be434d

xz.flip([-1]),