Hello, can this model handle 5D data of type (B, C, D, H, W)? - Githubissues

jacklishufan / Mamba-ND

Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data

46 stars 4 forks source link

Hello, can this model handle 5D data of type (B, C, D, H, W)? #4

Closed ILVHYY closed 3 months ago

ILVHYY commented 3 months ago

Hello, can this model handle 5D data of type (B, C, D, H, W)?

jacklishufan commented 3 months ago

Hi, is B the batch dimension and C the channel dimension? If so, then your data is actually 3D and we can handle it just like we can handle video data with size (B,C,T,H,W), where T is time