mlx.core.rsqrt

mlx.core.rsqrt(a: array, /, *, stream: Union[None, Stream, Device] = None) → array

Element-wise reciprocal and square root.

Parameters:

a (array) – Input array.

Returns:

One over the square root of a.

Return type:

array