rsqrt¶

nvtripy.rsqrt(input: Tensor) → Tensor[source]¶

Computes the elementwise reciprocal square root of the elements of the input tensor.

Parameters:: input (Tensor) – [dtype=T1] The input tensor.
Returns:: [dtype=T1] A new tensor of the same shape as the input tensor.
Return type:: Tensor

TYPE CONSTRAINTS:

T1: float8, float16, bfloat16, float32

Example

input = tp.arange(3, dtype=tp.float32) + 1.0
output = tp.rsqrt(input)

Local Variables¶

>>> input
tensor([1.0000, 2.0000, 3.0000], dtype=float32, loc=gpu:0, shape=(3,))

>>> output
tensor([1.0000, 0.7071, 0.5774], dtype=float32, loc=gpu:0, shape=(3,))