Transformer
class torch.nn.Transformer(d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048, dropout=0.1, activation=, custom_encoder=None, custom_decoder=None, layer_norm_eps=1e-05, batch_first=False, norm_first=False, device=None, dtype=None)
A transformer model. User is able to modify the attributes as
   needed. The architecture 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. In Advances in Neural
   Information Processing Systems, pages 6000-6010.
Parameters:
      * d_model (int) -- the number of expected features in the
        encoder/decoder inputs (default=512).
  * **nhead** (*int*) -- the number of heads in the
    multiheadattention models (default=8).

  * **num_encoder_layers** (*int*) -- the number of sub-encoder-
