LayerNorm — PyTorch 1.10.1 documentation
pytorch.org › docs › stableThe mean and standard-deviation are calculated over the last D dimensions, where D is the dimension of normalized_shape.For example, if normalized_shape is (3, 5) (a 2-dimensional shape), the mean and standard-deviation are computed over the last 2 dimensions of the input (i.e. input.mean((-2,-1))).
TransformerDecoderLayer — PyTorch 1.10.1 documentation
pytorch.org › docs › stableTransformerDecoderLayer is made up of self-attn, multi-head-attn and feedforward network. This standard decoder layer is based on the paper “Attention Is All You Need”. Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Lukasz Kaiser, and Illia Polosukhin. 2017. Attention is all you need.