How do I properly return an array after iterating and doing PCA

Question

I have a 3d array (sample, timestep, feature) named x_train in which I want to iterate through and perform PCA on the 2D array (timestep, feature) for every sample. I have this