快捷方式

LayerNorm

class torch.ao.nn.quantized.LayerNorm(normalized_shape, weight, bias, scale, zero_point, eps=1e-05, elementwise_affine=True, device=None, dtype=None)[source][source]

這是 LayerNorm 的量化版本。

額外引數
  • scale - 輸出的量化尺度,型別:double。

  • zero_point - 輸出的量化零點,型別:long。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

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

檢視教程

資源

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

檢視資源