How to convert human motion input of shape (num_frames, 21, 3) to (25, 6, num_frames) for into_critic

[divanoLetto]

Hi,
Thanks for sharing this great work!

I have a question about data formatting. I have human motion data with shape (num_frames, 21, 3) (21 joints, 3D positions over time), and I’d like to use it as input to the into_critic function, which seems to expect a tensor of shape (25, 6, num_frames).

Could you please clarify how I should convert or preprocess my data to match this expected input format?

Any help or example would be greatly appreciated.
Thanks in advance!