@simopal6 thank you. Yes , l think it is the simplest way to do that.
l’m also wandering why there are 4 values for means and 4 values for std  ?
mean=[0.485, 0.456, 0.406]
std=[0.229, 0.224, 0.225]

Thank you