Thank you @tqchen! I will try this approach and update here if I have further questions.
--- [Visit Topic](https://discuss.tvm.apache.org/t/axis-separator-in-relax-tensor/15385/10) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/8ed0225b52387d79ac11fbf21499621c0a1ed13ff5e86dea0f03268298f737de).