How to convert predicted trajectory of agents back to the coordinate of ego?

[GilgameshD]

I have one question about the visualization of prediction results. After I have the prediction of joint mode, how can I plot both ego and agents in the ego's map? I find that the predicted trajectory of agents are in their own coordinates.