torch.nn.functional.layer_norm

torch.nn.functional.layer_norm(input, normalized_shape, weight=None, bias=None, eps=1e-05)[源代码]

对最后几个维度应用层归一化。

详情请参阅LayerNorm

返回类型

Tensor

本页目录