Rsqrt

In-place version of Rsqrt.

Abstract Signature:

Rsqrt_(input: Tensor)

PyTorch

API: torch.rsqrt_
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.lax.rsqrt
Strategy: Direct Mapping
Official Docs ↗