pytorch中实现的层规范化
torch-layer-normalization的Python项目详细描述
pytorch层规范化
论文的实现:Layer Normalization
安装
pip install torch-layer-normalization
用法
fromtorch_layer_normalizationimportLayerNormalizationLayerNormalization(normal_shape=normal_shape)# The `normal_shape` could be the last dimension of the input tensor or the shape of the input tensor.