mlx.core.rsqrt

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

逐元素的倒数平方根。

参数:

a (array) – 输入数组。

返回值:

a 的平方根的倒数。

返回类型:

array