We should add overloads of mean, sum, stddev etc that take multiple dimensions. From the torch documentation for [mean](https://pytorch.org/docs/stable/generated/torch.mean.html): > "Returns the mean value of each row of the input tensor in the given dimension dim. If dim is a list of dimensions, reduce over all of them"