triton.language.sqrt_rn

triton.language.sqrt_rn(x)

计算 x 的逐元素的精确平方根（按照 IEEE 标准四舍五入到最近）。

参数:

x (Block) – 输入值