mlx_graphs.utils.scatter.invert_sqrt_degree

mlx_graphs.utils.scatter.invert_sqrt_degree(degree: mlx.core.array) → mlx.core.array

Computes the inverted square root of the degree array.

Parameters:

degree (array) – Array of length num_nodes with the inverted square root degree of each node.

Return type:

array

Returns:

Array of length num_nodes with the inverted square root of the degree of each node with ‘inf’ values zeroed out.