chickadee » nanograd » make-am-batch-norm

make-am-batch-norm num-features #!key (epsilon 1e-05) (momentum 0.1) (dtype 'f64)procedure

Batch normalization over the batch axis (axis 0), with learned gamma and beta of length num-features.