megengine.functional.clip

clip(x, lower=None, upper=None)[源代码]

将输入张量的所有元素限制在 [ lower, upper ] 区间并返回相应的结果:

\[\begin{split}y_i = \begin{cases} \text{lower} & \text{if } x_i < \text{lower} \\ x_i & \text{if } \text{lower} \leq x_i \leq \text{upper} \\ \text{upper} & \text{if } x_i > \text{upper} \end{cases}\end{split}\]
参数
  • x (Tensor) – (Tensor): 输入张量。

  • lower – (Numberic,optional): 限制区间的下界。

  • upper – (Numberic,optional): 限制区间的上界。

注解

  • 如果 lowerupper 都是 None, 触发 AssertionError.

  • 如果 lower 大于 upper, 结果和 clip(Tensor(), upper, upper) 相同。

返回类型

Tensor

返回

输出被限制区间后的张量。 返回结果的数据类型由 类型提升规则 决定。