AXIS_SEPARATOR in Relax Tensor

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