快捷方式

torch.nn.functional.rms_norm

torch.nn.functional.rms_norm(input, normalized_shape, weight=None, eps=None)[原始碼][原始碼]

應用均方根層正規化 (Root Mean Square Layer Normalization)。

有關詳細資訊,請參閱 RMSNorm

傳回類型

Tensor

文件

取得 PyTorch 的全面開發者文件

查看文件

教學

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

查看教學

資源

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

檢視資源