【Triton 教程】triton_language.clamp

Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境，以高效编写自定义 DNN 计算内核，并能够在现代 GPU 硬件上以最大吞吐量运行。

*在线运行 Triton 学习教程 → https://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，则最终结果将无法确定。