Pytorch Transformer Encoder Block