项目作者： CyberZHG

项目描述：
Layer normalization in PyTorch

高级语言： Python

项目主页： https://pypi.org/project/torch-layer-normalization/

项目地址: git://github.com/CyberZHG/torch-layer-normalization.git

创建时间： 2019-02-18T06:34:01Z
项目社区：https://github.com/CyberZHG/torch-layer-normalization
开源协议：MIT License
下载

PyTorch Layer Normalization

Implementation of the paper: Layer Normalization

Install

pip install torch-layer-normalization

Usage

from torch_layer_normalization import LayerNormalization
LayerNormalization(normal_shape=normal_shape)
# The `normal_shape` could be the last dimension of the input tensor or the shape of the input tensor.