I meet another error at evaluation step2 and step3, because of len(prediction_trajectory.shape) != 4. 
I meet another error at evaluation step2 and step3, because of len(prediction_trajectory.shape) != 4.
