評價此頁

hardswish#

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

這是 hardswish() 的量化版本。

引數
  • input (Tensor) – 量化輸入

  • scale (float) – 輸出張量的量化尺度

  • zero_point (int) – 輸出張量的量化零點

返回型別

張量