nn.LayerNorm()中的参数

nn.LayerNorm()中的参数包括：
- normalized_shape：表示输入张量的形状，可以是一个整数或一个整数列表或一个torch.Size对象。
- eps：表示在标准差的计算中添加到方差的值，以避免除以零的情况，默认值为1e-5。
- elementwise_affine：表示是否对输出进行仿射变换，默认为True，即进行仿射变换。

例如，对于一个形状为[3,4]的张量a，可以使用以下代码创建一个LayerNorm层并将a传递给它进行标准化：

import torch
import torch.nn as nn

a = torch.randint(10, [3, 4]).to(torch.float32)
layer_norm = nn.LayerNorm([3, 4])
output = layer_norm(a)