Python wrappers
Public Python APIs implemented by this engine symbol.Autograd node for the element-wise reciprocal square root .
Saves the output so the backward pass can form without
re-evaluating rsqrt. Defined for .
Math
\frac{\partial y}{\partial x} = -\tfrac{1}{2}\,x^{-3/2} = -\tfrac{1}{2}\,y^3, \qquad \frac{\partial \mathcal{L}}{\partial x} = -\tfrac{1}{2}\,y^3\,\frac{\partial \mathcal{L}}{\partial y}, \quad x > 0.$$Attributes
schema_v1OpSchemaRegistered as
"rsqrt" with AmpPolicy::Promote.kSavesInputboolfalse.kSavesOutputbooltrue — saved tensor is .Notes
Dispatch: Accelerate vvrsqrtf/vvrsqrt (CPU) / MLX rsqrt (GPU).