Module batchnorm

Module batchnorm 

Source

Functionsยง

batch_norm_1d_inference
BatchNorm1D Inference Implementation (CPU Naive) Uses running stats instead of batch stats.
batch_norm_1d_training
BatchNorm1D Training Implementation (CPU Naive)
batchnorm_backward_naive
BatchNorm Backward Naive (Training Mode) Returns (grad_input, grad_weight, grad_bias)