layers in the encoder (default=6).
  * **num_decoder_layers** (*int*) -- the number of sub-decoder-
    layers in the decoder (default=6).

  * **dim_feedforward** (*int*) -- the dimension of the
    feedforward network model (default=2048).

  * **dropout** (*float*) -- the dropout value (default=0.1).

  * **activation** (*Union**[**str**,
    **Callable**[**[**Tensor**]**, **Tensor**]**]*) -- the
    activation function of encoder/decoder intermediate layer, can
    be a string ("relu" or "gelu") or a unary callable. Default:
    relu

  * **custom_encoder** (*Optional**[**Any**]*) -- custom encoder
    (default=None).

  * **custom_decoder** (*Optional**[**Any**]*) -- custom decoder
    (default=None).

  * **layer_norm_eps** (*float*) -- the eps value in layer
    normalization components (default=1e-5).

  * **batch_first** (*bool*) -- If "True", then the input and
