triton.language.rsqrt

triton.language.rsqrt(x, _semantic=None)

Computes the element-wise inverse square root of x.

Parameters:

x (Block) – the input values

This function can also be called as a member function on tensor, as x.rsqrt() instead of rsqrt(x).