【Triton 教程】triton_language.clamp

0 阅读1分钟

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有效。如果minmax中任一值为NaN,则最终结果将无法确定。