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 – 输入张量。

  • lower – 限定区间的下界

  • upper – 限定区间的上界

返回

输出经过区间限定的张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

a = tensor(np.arange(5).astype(np.int32))
print(F.clip(a, 2, 4).numpy())
print(F.clip(a, lower=3).numpy())
print(F.clip(a, upper=3).numpy())

输出:

[2 2 2 3 4]
[3 3 3 3 4]
[0 1 2 3 3]