triton.language.clamp

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

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

参数:

• x (Block) – 输入张量

• min (Block) – 限制的下界

• max (Block) – 限制的上界

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

另请参阅

tl.PropagateNan