mlx.core.rsqrt

rsqrt(a: array, /, *, stream: None | Stream | Device = None) → array

Element-wise reciprocal and square root.

參數:

a (array) -- 輸入陣列。

回傳值:

One over the square root of a.

回傳值型別:

array