How do we calculate channel wise normalization stats?

How do we calculate channel wise normalization stats? What kind of function should we write to obtain the mean and standard deviation on a custom dataset?

1 Like