triton.language.clamp

triton.language.clamp(x, min, max, propagate_nan: ~triton.language.core.constexpr = <PROPAGATE_NAN.NONE: 0>)

将输入张量x限制在[min, max]范围内。 当min大于max时的行为是未定义的。

Parameters:

• x (Block) – 输入张量

• min (Block) – 钳位的下限

• max (Block) – 钳位的上限值

• propagate_nan (tl.PropagateNan) – 是否传播NaN值。仅适用于x张量。 如果min或max中有一个是NaN，结果将是未定义的。

另请参阅

tl.PropagateNan