評價此頁

torch.nn.functional.layer_norm#

torch.nn.functional.layer_norm(input, normalized_shape, weight=None, bias=None, eps=1e-05)[原始碼]#

對最後特定數量的維度應用層歸一化。

詳情請參見 LayerNorm

返回型別

張量