Triton
入门
安装
教程
Python API
triton
triton.language
编程模型
创建操作 (Creation Ops)
形状操作 (Shape Manipulation Ops)
线性代数操作 (Linear Algebra Ops)
内存/指针操作 (Memory/Pointer Ops)
索引操作 (Indexing Ops)
数学操作 (Math Ops)
triton.language.abs
triton.language.cdiv
triton.language.ceil
triton.language.clamp
triton.language.cos
triton.language.div_rn
triton.language.erf
triton.language.exp
triton.language.exp2
triton.language.fdiv
triton.language.floor
triton.language.fma
triton.language.log
triton.language.log2
triton.language.maximum
triton.language.minimum
triton.language.rsqrt
triton.language.sigmoid
triton.language.sin
triton.language.softmax
triton.language.sqrt
triton.language.sqrt_rn
sqrt_rn()
triton.language.umulhi
归约操作 (Reduction Ops)
扫描/排序操作 (Scan/Sort Ops)
原子操作 (Atomic Ops)
随机数生成
迭代器
内联汇编
编译器提示操作 (Compiler Hint Ops)
调试操作 (Debug Ops)
triton.testing
Triton 语义
Triton MLIR 方言
Triton MLIR 方言和操作
编程指南
简介
相关工作
调试 Triton
Triton
triton.language
triton.language.sqrt_rn
查看页面源
triton.language.sqrt_rn
¶
triton.language.
sqrt_rn
(
x
,
_semantic
=
None
)
¶
计算
x
的逐元素精确平方根(根据 IEEE 标准四舍五入到最近的值)。
参数
:
x
(
Block
) – 输入值