PytorchBackend.mean

PytorchBackend.mean(*args, **kwargs)

Compute the mean of array elements.

Parameters:

arrBackendArray

Input data.

axisint or tuple of ints, optional

Axis or axes to perform the operation on. Default is all.

Returns:

Union[BackendArray, Scalar]

Mean of arr.