mean and std are usually containing values for each channel, so that each channel will be normalized separately.
If you are dealing with e.g. 10 channels, mean and std should be a tensor containing 10 values.