PytorchBackend.center_of_mass

PytorchBackend.center_of_mass(arr, cutoff=None)

Computes the center of mass of an array larger than cutoff.

Parameters:

arrBackendArray

Array to compute the center of mass of.

cutofffloat, optional

Densities less than or equal to cutoff are nullified for center of mass computation. Defaults to None.

Returns:

BackendArray

Center of mass with shape (arr.ndim).