cRSE with more than 3 dim

[hpc100]

Thank's for sharing your work,

I wonder if it's possible to pass 4 dim features to your cRSE function. And if so which code I need to modify ? (self_attn_aio_fwd.cu ? or other).

Thank's for any feedback