triton.language.clamp

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

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

参数:

• x (Block) – 输入张量

• min (Block) – 限制的下限

• max (Block) – 限制的上限

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

另请参阅

tl.PropagateNan