Calculates the mean of elements across dimensions of a tensor.
the input tensor
Optional
(optional) the dimension(s) to reduce. By default it reduces all dimensions.
(optional) if true, retains reduced dimensions with size 1.
the mean
Calculates the mean of elements across dimensions of a tensor.