triton.language.sqrt_rn

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

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

参数:

x (Block) – 输入值