钳位

torch.ao.nn.quantized.functional.clamp(input, min_, max_)[源代码]

float(input, min_, max_) -> Tensor

按元素应用钳位函数。详情参见clamp

参数
  • input (Tensor) – 输入张量的量化版本

  • min - 设置夹紧的最小值

  • max - 用于上限约束的最大值

返回类型

Tensor

本页目录