快捷方式

hardswish

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

這是 hardswish() 的量化版本。

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

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

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

回傳類型

Tensor

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並取得您的問題解答

檢視資源