评价此页

hardswish#

class torch.ao.nn.quantized.functional.hardswish(input, scale, zero_point)[source]#

这是 hardswish() 的量化版本。

参数
  • input (Tensor) – 量化输入

  • scale (float) – 输出张量的量化尺度

  • zero_point (int) – 输出张量的量化零点

返回类型

张量