Inference runs on one example at a time, so batch statistics are unavailable or unstable. Batch norm instead uses running averages collected during training, fixed at inference time.
Implement batchnorm_inference(x, gamma, beta, running_mean, running_var, eps), which normalizes with the stored running statistics.
running_mean and running_var have shape (D,).Use running_mean and running_var in place of the batch mean and variance, then scale and shift.
No results yet
Run your code to check it against the examples, or submit to run every test.