batch[:, c] will index the current channel and torch.mean as well as torch.std will calculate the mean and standard deviation for this channel, respectively.
How should the result of the stats “across the channels” look like, i.e. which shape should it have?
If you want to calculate the mean in the channel dimension, you could use torch.mean(batch, dim=1).