torch:mean

(torch:mean a &key axis keepdims)

Differentiable arithmetic mean (linalg:mean, same :axis / :keepdims rules as torch:sum); the backward pass is the sum adjoint divided by the reduced element count. The mean of a squared torch:sub is the MSE loss.