Confusing cuda shape

[niconiconi]

https://github.com/PolyhedraZK/ExpanderCompilerCollection/blame/260f4919e7049023e26b0b295eba3ace2b232dbd/docs/docs/cuda/cuda_like_frontend.md#L220

[niconiconi]

Ideally you can use torch style array, where it's actually a flattened array but has all kinds of shapes.

https://github.com/PolyhedraZK/ExpanderCompilerCollection/blame/260f4919e7049023e26b0b295eba3ace2b232dbd/docs/docs/cuda/cuda_like_frontend.md#L216