AXIS_SEPARATOR in Relax Tensor

[abhikran-quic]

Thank you @tqchen! I will try this approach and update here if I have further questions.