hustvl / Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Apache License 2.0
2.87k stars 185 forks source link

About data flipping #101

Open zzzendurance opened 3 months ago

zzzendurance commented 3 months ago

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 commented 3 months ago

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