Calculates the mean of elements across dimensions of a tensor.
the mean
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.
Generated using TypeDoc
Calculates the mean of elements across dimensions of a tensor.
Returns
the mean