mlx_graphs.utils.scatter.scatter_max

mlx_graphs.utils.scatter.scatter_max(src: mlx.core.array, index: mlx.core.array, values: mlx.core.array)

Scatters values at index within src. If duplicate indices are present, the maximum value is kept at these indices.

Parameters:

• src (array) – Source array where the values will be scattered (often an empty array)

• index (array) – Array containing indices that determine the scatter of the ‘values’.

• values (array) – Input array containing values to be scattered.

Returns:

The resulting array after applying scatter and max operations on the values at duplicate indices