triton.language.sqrt_rn

triton.language.sqrt_rn(x)

计算x逐元素的精确平方根(根据IEEE标准四舍五入到最近值)。

Parameters:

x (Block) – 输入值