快捷方式

hardswish

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

這是量化版本的 hardswish()

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

  • scale (float) – 輸出張量的量化比例

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

返回型別

Tensor

文件

獲取 PyTorch 全面的開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源