If you want to keep the image tensors in the grayscale format, you could use the Normalize transformation from my previous post with a single mean and std value and it should work.
Currently you are passing 3 values for these stats, which will most likely yield a shape mismatch error.