Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境,以高效编写自定义 DNN 计算内核,并能够在现代 GPU 硬件上以最大吞吐量运行。
*在线运行 Triton 学习教程 → go.hyper.ai/wS9x1
triton.language.clamp(x, min, max, propagate_nan: ~triton.language.core.constexpr = <PROPAGATE_NAN.NONE: 0>)
将输入张量 x 的值限制在 [min, max] 范围内。
参数:
- x (Block) - 输入值。
- min (Block) – 限制操作的下界值。
- max (Block) – 限制操作的上界值。
- propagate_nan (tl.PropagateNan) – 是否将 NaN 值传播出去。此设置仅对张量
x有效。如果min或max中任一值为NaN,则最终结果将无法确定。