mlx.core.round#

mlx.core.round(a: array, /, decimals: int = 0, stream: Union[None, Stream, Device] = None) → array#

Round to the given number of decimals.

Basically performs:

s = 10**decimals
x = round(x * s) / s
Parameters:

  • a (array) – Input array

  • decimals (int) – Number of decimal places to round to. (default: 0)

Returns:

An array of the same type as a rounded to the given number of decimals.

Return type:

result (array)