The idea of augmenting data is generating ``similar" samples from the data generating distribution possibly to avoid overfitting. The values of mean and std_dev will change however, I believe the difference wouldn’t be significant. You can go ahead and test this hypothesis by computing mean and std_dev after transforms. Also, you apply the same transformation on the test set. Your reasoning is correct in this regard.