Shortcuts

Hardswish

class torch.ao.nn.quantized.Hardswish(scale, zero_point, device=None, dtype=None)[源代码]

这是 Hardswish 的量化版本。

Parameters
  • scale – 输出张量的量化尺度

  • zero_point – 输出张量的量化零点

优云智算