Shortcuts

hardswish

class torch.ao.nn.quantized.functional.hardswish(input, scale, zero_point)[源代码]

这是 hardswish() 的量化版本。

Parameters
  • 输入 (张量) – 量化输入

  • scale (float) – 输出张量的量化比例

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

Return type

张量

优云智算