thank you for answering to my question!
I followed your suggestion and
Normalize function works
as I passed each [3,32,32] image (not batch together [128,3,32,32])
for I in range(data.shape[0]):
    data[i]  = transforms.Normalize(mean,std)(data[i])