Simplify mean and std computation

The mean/std computation is performed here but is overly complicated.

It should be simplified, see the Doc-UFCN computation for example: https://gitlab.com/teklia/dla/doc-ufcn/-/blob/main/doc_ufcn/train/normalization_params.py#L20