Could you tell me the expected shape of bone_a, x_a, x_a1, xp_a in the forward() function of IPES_Trainer() ?